验证二叉树的前序序列化

题目链接

题目：

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

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

代码：

class Solution:
    def isValidSerialization(self, preorder: str) -> bool:
        preorder=preorder.split(',')
        nodeCount,leafCount=0,0
        for node in preorder:
            if leafCount-nodeCount>0:
                return False
            if node=='#':
                leafCount+=1
            else :
                nodeCount+=1
        return leafCount-nodeCount==1

题解：

把‘#’看作叶子节点，由二叉树的性质可知，
①叶子节点的个数=非叶子节点的个数+1.
②而在先序遍历中，遍历到的最后一个节点一定是叶子节点，所以在前序遍历中（遍历到最后一个叶子节点之前），非叶子节点一定大于等于叶子节点。
所以,遍历前序序列，遇到“#”则把叶子节点数+1，否则，把非叶子节点数+1.如果在遍历到最后一个元素之前，叶子节点数大于非叶子节点数，由②知，一定不是二叉树。
最后，遍历完前序序列，如果叶子节点数不等于非叶子节点数+1，由①知，一定不是二叉树。
否则就是二叉树。