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

解答：

从序列起始处开始遍历,如果遇到数值就入栈，如果遇到"#“号,说明该位置已经到达了叶子节点,这时候需要查看栈顶是否也为”#",如果是,说明栈顶的前一位置就是该叶子节点的父节点.因此我们将栈顶元素和栈顶的上一元素出栈,然后用"#“替代(入栈,表示某一分支已经结束),这样操作后,最终根节点的左右孩子都会被替代为”#"，合并后栈中只剩下一个"#"，表明所有节点都已经遍历完。如果最后栈中大小不为1，说明该序列是错误的

class Solution(object):
    def isValidSerialization(self, preorder):
        stack = []
        for node in preorder.split(','):
            stack.append(node)
            while len(stack) >= 3 and stack[-1] == stack[-2] == '#' and stack[-3] != '#':
                stack.pop(), stack.pop(), stack.pop()
                stack.append('#')
        return len(stack) == 1 and stack.pop() == '#'