0331：验证二叉树的前序序列化（★）

力扣第 331 题

题目

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

例如，上面的二叉树可以被序列化为字符串 "9,3,4,#,#,1,#,#,2,#,6,#,#"，其中 # 代表一个空节点。

给定一串以逗号分隔的序列，验证它是否是正确的二叉树的前序序列化。编写一个在不重构树的条件下的可行算法。

保证 每个以逗号分隔的字符或为一个整数或为一个表示 null 指针的 '#' 。

你可以认为输入格式总是有效的

• 例如它永远不会包含两个连续的逗号，比如 "1,,3" 。

注意：不允许重建树。

示例 1:

输入: preorder = "9,3,4,#,#,1,#,#,2,#,6,#,#"
输出: true

示例 2:

输入: preorder = "1,#"
输出: false

示例 3:

输入: preorder = "9,#,#,1"
输出: false

提示:

• 1 <= preorder.length <= 104
• preorder 由以逗号 “，” 分隔的 [0,100] 范围内的整数和 “#” 组成

分析

• 注意到叶子节点后必然是两个空节点
• 不断将叶子节点和后面的两个空节点替换为一个空节点
• 直到只剩下一个空节点，即说明是正确的

解答

1
2
3
4
5
6
7
8
9

class Solution:
    def isValidSerialization(self, preorder: str) -> bool:
        sk = []
        for c in preorder.split(','):
            while len(sk)>1 and c==sk[-1]=='#'!=sk[-2]:
                sk.pop()
                sk.pop()
            sk.append(c)
        return sk==['#']

33 ms