题目:
序列化二叉树的一种方法是使用 前序遍历 。当我们遇到一个非空节点时,我们可以记录下这个节点的值。如果它是一个空节点,我们可以使用一个标记值记录,例如 #
。
例如,上面的二叉树可以被序列化为字符串 "9,3,4,#,#,1,#,#,2,#,6,#,#"
,其中 #
代表一个空节点。
给定一串以逗号分隔的序列,验证它是否是正确的二叉树的前序序列化。编写一个在不重构树的条件下的可行算法。
保证 每个以逗号分隔的字符或为一个整数或为一个表示 null
指针的 '#'
。
你可以认为输入格式总是有效的
- 例如它永远不会包含两个连续的逗号,比如
"1,,3"
。
注意:不允许重建树。
思考:
1. 用“#”表示空节点,那么所有值不为“#”的节点都有两个子节点 ----> 用一个数据结构记录每个节点的两个子节点空位是否被填充
2. 前序遍历二叉树 ----> 先进后出 ----> 栈
代码如下:
class Solution(object):
def isValidSerialization(self, preorder):
"""
:type preorder: str
:rtype: bool
"""
nodes = preorder.split(',') # 按逗号分割字符串
stack = [1] # 用栈存储节点的子节点左右槽位,初始化
for node in nodes:
if not stack: # 若栈空,即还有节点没有填充,但已经没有槽位,返回False
return False
stack[-1] -= 1 # 否则填充节点,将栈顶槽位减一
if stack[-1] == 0: # 若栈顶为0,即该节点槽位已满
stack.pop()
if node != '#':
# 对于普通节点,增加一个值为2的槽位
stack.append(2)
if not stack: # 栈空说明整棵树填充满了,证明是二叉树
return True
else:
return False
提交通过: