1028. 从先序遍历还原二叉树
难度:困难
时间:2023/04/22
我们从二叉树的根节点 root 开始进行深度优先搜索。
在遍历中的每个节点处,我们输出 D 条短划线(其中 D 是该节点的深度),然后输出该节点的值。(如果节点的深度为 D,则其直接子节点的深度为 D + 1。根节点的深度为 0)。
如果节点只有一个子节点,那么保证该子节点为左子节点。
给出遍历输出 S,还原树并返回其根节点 root。
示例 1:
输入:"1-2--3--4-5--6--7"
输出:[1,2,5,3,4,6,7]
示例 2:
输入:"1-2--3---4-5--6---7"
输出:[1,2,5,3,null,6,null,4,null,7]
示例 3:
输入:"1-401--349---90--88"
输出:[1,401,null,349,88,90]
提示:
- 原始树中的节点数介于
1和1000之间。 - 每个节点的值介于
1和10 ^ 9之间。
这道题最关键的是捋清楚深度和父子节点之间的关系。很容易发现如果A为B的子节点,那么DepthA>DepthB。 所以我们只需要对S字符串进行全局遍历即可。同一个全局变量cur去处理节点值,同时用一个全局变量curD维护当前遍历到的深度,再用一个栈变量depth维护当前节点深度,如果curD>depth则显然全局遍历的下一节点为当前节点的子节点,否则为其长辈节点(叔父节点,祖叔父节点...)。
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def recoverFromPreorder(self, traversal: str) -> TreeNode:
path, pos = list(), 0
while pos < len(traversal):
level = 0
while traversal[pos] == '-':
level += 1
pos += 1
value = 0
while pos < len(traversal) and traversal[pos].isdigit():
value = value * 10 + (ord(traversal[pos]) - ord('0'))
pos += 1
node = TreeNode(value)
if level == len(path):
if path:
path[-1].left = node
else:
path = path[:level]
path[-1].right = node
path.append(node)
return path[0]