Leetcode 1028.从先序遍历还原二叉树
1 题目描述(Leetcode题目链接)
我们从二叉树的根节点 root 开始进行深度优先搜索。
在遍历中的每个节点处,我们输出 D 条短划线(其中 D 是该节点的深度),然后输出该节点的值。(如果节点的深度为 D,则其直接子节点的深度为 D + 1。根节点的深度为 0)。
如果节点只有一个子节点,那么保证该子节点为左子节点。
给出遍历输出 S,还原树并返回其根节点 root。
输入:"1-2--3--4-5--6--7"
输出:[1,2,5,3,4,6,7]
提示:
- 原始树中的节点数介于 1 和 1000 之间。
- 每个节点的值介于 1 和 10 ^ 9 之间。
2 题解
用一个栈来存节点。
class Solution:
def recoverFromPreorder(self, S: str) -> TreeNode:
stack = []
i = 0
while i < len(S):
num = 0
val = ""
while S[i] == "-":
num += 1
i += 1
while i < len(S) and S[i] != "-":
val += S[i]
i += 1
node = TreeNode(int(val))
if not stack:
stack.append(node)
continue
if num == len(stack):
stack[-1].left = node
else:
while num != len(stack):
stack.pop()
stack[-1].right = node
stack.append(node)
return stack[0]
记录一个用正则表达式做的方法,思路应该是一样的。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def recoverFromPreorder(self, S: str) -> TreeNode:
i = S.find('-')
if i == -1:
return TreeNode(int(S)) if S else None
stack = [TreeNode(int(S[:i]))]
for d, num in re.findall(r'(\-+)(\d+)', S):
del stack[len(d):]
node = TreeNode(int(num))
if not stack[-1].left:
stack[-1].left = node
else:
stack[-1].right = node
stack.append(node)
return stack[0]