序列化二叉树的一种方法是使用 前序遍历 。当我们遇到一个非空节点时,我们可以记录下这个节点的值。如果它是一个空节点,我们可以使用一个标记值记录,例如 #
。
例如,上面的二叉树可以被序列化为字符串 "9,3,4,#,#,1,#,#,2,#,6,#,#",其中 # 代表一个空节点。
给定一串以逗号分隔的序列,验证它是否是正确的二叉树的前序序列化。编写一个在不重构树的条件下的可行算法。
保证 每个以逗号分隔的字符或为一个整数或为一个表示 null 指针的 '#' 。
你可以认为输入格式总是有效的
例如它永远不会包含两个连续的逗号,比如 "1,,3" 。
注意:不允许重建树。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/verify-preorder-serialization-of-a-binary-tree
例:
输入: preorder = "9,3,4,#,#,1,#,#,2,#,6,#,#" 输出: true
解析:
可以将‘#’符号看做叶子节点,而叶子节点相较于其他节点要多一。所以可以设置一个值初始为1,每增加一个非叶子结点就+1,而每出现一个叶子节点(#)就减一,如果符合前序遍历的规则,那么在遍历完之前不会为0,因为非叶子节点一定多余叶子节点(#)。而遍历完后一定为0,因为叶子节点(#)只多一个。
class Solution(object):
def isValidSerialization(self, preorder):
"""
:type preorder: str
:rtype: bool
"""
flag = 1 # 判断值
for node in preorder.split(','): # 将字符串中的','剔除。
if flag == 0: # 遍历完之前达到0说明不符合前序遍历规则
return False
if node == '#': # “叶子节点”
flag -= 1
else: # “非叶子节点”
flag += 1
return flag == 0 # 遍历完后的判断