【LeetCode-331】验证二叉树的前序序列化

问题

序列化二叉树的一种方法是使用前序遍历。当我们遇到一个非空节点时，我们可以记录下这个节点的值。如果它是一个空节点，我们可以使用一个标记值记录，例如 #。

示例

输入： "9,3,4,#,#,1,#,#,2,#,6,#,#"
输出： true

解答

class Solution {
public:
    bool isValidSerialization(string preorder) {
        vector<char> v;
        preorder += ',';
        int n = preorder.size();
        for (int i = 1; i < n ; i++) {
            if (preorder[i] != ',') continue;
            v.push_back(preorder[i - 1]);
            while (check(v)) { // 删除一个叶子结点
                v.pop_back();
                v.pop_back();
                v.back() = '#';
            }
        }
        return v == vector<char>{'#'};
    }
    bool check(vector<char>& v) {
        int n = v.size() - 1;
        if (n > 1 && v[n] == '#' && v[n - 1] == '#' && isdigit(v[n - 2])) // 前序为“数##”
            return true;
        return false;
    }
};

重点思路

从左到右遍历，消消乐的思想，对于前序遍历，遇到num, #, #这样的形式，说明到达了一个叶子节点，把它替换为#，也就是替换为空节点，继续比较。最后看是不是只剩下#。这个思想对于前中后都适用，只需要改一下check函数即可。