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

解答

由于题目中给出的序列化方案将所有空节点都计入，所以#实际上等价于叶子节点，可以发现在这种序列化方案下，叶子节点的数目总是等于非叶子节点数目加一，因此而可以统计字符串中#的数目是否等价于数字数目加一：

class Solution {
public:
    bool isValidSerialization(string preorder) {
        int len = preorder.size();
        if(preorder[0] == '#')
            return len == 1;
        // 记录 # 数量，由于没有统计第一位数字，所以初始化为-1
        int cnt = -1;
        for(int i = 1; i < len - 1; i++){
            if(preorder[i] == ','){
                if(preorder[i + 1] == '#'){
                    cnt++;
                }
                else{
                    cnt--;
                }
            }
            if(cnt > 0 && i < len - 2)
                return false;

        }
        return cnt == 1;
    }
};