# leetcode 331.验证二叉树的前序序列化

leetcode 331.验证二叉树的前序序列化

题干

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

# # # #

例如，上面的二叉树可以被序列化为字符串 “9,3,4,#,#,1,#,#,2,#,6,#,#”，其中 # 代表一个空节点。
给定一串以逗号分隔的序列，验证它是否是正确的二叉树的前序序列化。编写一个在不重构树的条件下的可行算法。
每个以逗号分隔的字符或为一个整数或为一个表示 null 指针的 ‘#’ 。
你可以认为输入格式总是有效的，例如它永远不会包含两个连续的逗号，比如 “1,3” 。

示例 1:
输入: “9,3,4,#,#,1,#,#,2,#,6,#,#”
输出: true

示例 2:
输入: “1,#”
输出: false

示例 3:
输入: “9,#,#,1”
输出: false

题解

贪心地考虑，遍历到的每个元素会占用前驱元素的一个孩子（即优先朝左侧添加孩子）
如果遍历到的元素是#则这个节点不会再生成分支；如果遍历到的元素不是#，则这个节点还会再生出两个分支
然后就可以想到用一个栈来维护当前节点还有几个孩子是空的（因为如果是空元素一定要有#），每加入一个非空元素便在推一个2入栈，表示当前节点有两个孩子可用，同时使此前栈顶元素-1表示消耗了一个孩子。如果遍历到空元素则不推入新的2。
注意边界判断即可。

class Solution {
public:
    bool isValidSerialization(string preorder) {
        int n = preorder.length();
        if(n == 0) return false;
        stack<int> stk;
        for(int i = 0 ; i < n ; ++i){
            if(preorder[i] == ',') continue;
            while(!stk.empty() && stk.top() == 0) stk.pop();
            if(stk.empty() && i != 0) return false;
            if(preorder[i] == '#'){
                //节点为空
                if(stk.empty()) continue;
                else stk.top()--;
            }else{
                //节点非空
                if(i == 0) stk.push(2);
                else{
                    stk.top()--;
                    stk.push(2);
                }
            }
        }
        if(!stk.empty() && stk.top() != 0) return false;
        return true;
    }
};