每日一题 验证二叉树的前序序列化
序列化二叉树的一种方法是使用 前序遍历 。当我们遇到一个非空节点时,我们可以记录下这个节点的值。如果它是一个空节点,我们可以使用一个标记值记录,例如 #
。
例如,上面的二叉树可以被序列化为字符串 "9,3,4,#,#,1,#,#,2,#,6,#,#"
,其中 #
代表一个空节点。
给定一串以逗号分隔的序列,验证它是否是正确的二叉树的前序序列化。编写一个在不重构树的条件下的可行算法。
保证 每个以逗号分隔的字符或为一个整数或为一个表示 null
指针的 '#'
。
你可以认为输入格式总是有效的
- 例如它永远不会包含两个连续的逗号,比如
"1,,3"
。
注意:不允许重建树。
解题思路
有两种思路,一种是看二叉树能否成功初始化;一种是在遍历字符串的过程中,用一个栈保存当前节点还有多少棵子树,遇到“#”子树(当前栈顶)-1,遇到数字子树-1,同时在栈中压入一个2(产生了一个新的节点,又有两个子树)。
代码实现
class Solution:
def isValidSerialization(self, preorder: str) -> bool:
stack = [1]
node_list = preorder.split(',')
for node in node_list:
if stack == []:
return False
if node == '#':
stack[-1] -= 1
if stack[-1] == 0:
stack.pop()
else:
stack[-1] -= 1
if stack[-1] == 0:
stack.pop()
stack.append(2)
return not stack
复杂度分析
时间:O(n)
空间:O(n)