根据一棵树的前序遍历与中序遍历构造二叉树。
注意:你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
3
/ \
9 20
/ \
15 7
方法一:递归【O(n),其中 n 是树中的节点个数】
class Solution:
def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
def myBuildTree(preorder_left: int, preorder_right: int, inorder_left: int, inorder_right: int):
if preorder_left > preorder_right:
return None
# 前序遍历中的第一个节点就是根节点
preorder_root = preorder_left
# 在中序遍历中定位根节点
inorder_root = index[preorder[preorder_root]]
# 先把根节点建立出来
root = TreeNode(preorder[preorder_root])
# 得到左子树中的节点数目
size_left_subtree = inorder_root - inorder_left
# 递归地构造左子树,并连接到根节点
# 先序遍历中「从 左边界+1 开始的 size_left_subtree」个元素就对应了中序遍历中「从 左边界 开始到 根节点定位-1」的元素
root.left = myBuildTree(preorder_left + 1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1)
# 递归地构造右子树,并连接到根节点
# 先序遍历中「从 左边界+1+左子树节点数目 开始到 右边界」的元素就对应了中序遍历中「从 根节点定位+1 到 右边界」的元素
root.right = myBuildTree(preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right)
return root
n = len(preorder)
# 构造哈希映射,帮助我们快速定位根节点
index = {
element: i for i, element in enumerate(inorder)}
return myBuildTree(0, n - 1, 0, n - 1)
方法二:递归
1.规律
前序:(root.val)[左子树的结点的数值们,长度为left_len] [右子树的结点的数值们,长度为right_len]
中序:[左子树的结点的数值们,长度为left_len] (root.val)[右子树的结点的数值们,长度为right_len]
idx = inorder.index(root.val) 就是左子树的结点个数
2.切片比较好用。用指针容易思维搞乱,且边界不好维护
# 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 buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
# 递归结束条件
if not preorder:
return None
root_val = preorder[0]
root_idx = inorder.index(root_val)
root = TreeNode(root_val)
# 前序遍历中,索引root_idx前的节点为根+左子树,preorder[1: root_idx + 1]为左子树数组
# 中序遍历中,索引root_idx为根节点,inorder[0: root_idx]为左子树数组
root.left = self.buildTree(preorder[1:root_idx + 1], inorder[0:root_idx])
# 索引root_idx后的节点为右子树所有节点,preorder[root_idx + 1:]为右子树数组
# 中序遍历中,索引root_idx为根节点,inorder[root_idx + 1: ]为右子树数组
root.right = self.buildTree(preorder[root_idx + 1:], inorder[root_idx + 1:])
return root
方法三:迭代【O(n),其中 n 是树中的节点个数】
class Solution:
def buildTree(self, preorder: List[int], inorder: List[int]) -> TreeNode:
if not preorder:
return None
root = TreeNode(preorder[0])
stack = [root]
inorderIndex = 0
for i in range(1, len(preorder)):
preorderVal = preorder[i]
node = stack[-1]
if node.val != inorder[inorderIndex]:
node.left = TreeNode(preorderVal)
stack.append(node.left)
else:
while stack and stack[-1].val == inorder[inorderIndex]:
node = stack.pop()
inorderIndex += 1
node.right = TreeNode(preorderVal)
stack.append(node.right)
return root
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/construct-binary-tree-from-preorder-and-inorder-traversal/solution/cong-qian-xu-yu-zhong-xu-bian-li-xu-lie-gou-zao-9/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。