106. 从中序与后序遍历序列构造二叉树
每日几道leetcode刷刷题!
传送门
题目描述
给定两个整数数组 inorder 和 postorder ,其中 inorder 是二叉树的中序遍历, postorder 是同一棵树的后序遍历,请你构造并返回这颗 二叉树 。
示例 1:
输入:inorder = [9,3,15,20,7], postorder = [9,15,7,20,3]
输出:[3,9,20,null,null,15,7]
代码
# 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, inorder: List[int], postorder: List[int]) -> TreeNode:
# 第一步: 特殊情况讨论: 树为空. (递归终止条件)
if not postorder:
return None
# 第二步: 后序遍历的最后一个就是当前的中间节点.
root_val = postorder[-1]
root = TreeNode(root_val)
# 第三步: 找切割点的前序的索引.(C++需要通过循环去查找)
root_index = inorder.index(root_val)
#第四步: 切割inorder数组. 得到inorder数组的左,右半边.
inorder_left = inorder[:root_index]
inorder_right = inorder[root_index + 1:]
# 难点:第五步: 切割postorder数组. 得到postorder数组的左,右半边
#中序数组大小一定跟后序数组大小是相同的.
postorder_left = postorder[:len(inorder_left)]
postorder_right = postorder[len(inorder_left): len(postorder) - 1]
# 第六步: 递归
root.left = self.buildTree(inorder_left, postorder_left)
root.right = self.buildTree(inorder_right, postorder_right)
return root
总结
来源于【代码随想录】
中序: 左子树 - 根 - 右指数
后序: 左子树 - 右子树 - 根
⾸先回忆⼀下如何根据两个顺序构造⼀个唯⼀的⼆叉树,相信理论知识⼤家应该都清楚,就是以 后序数组
的最后⼀个元素
为切割点
(根),先切中序数组,根据中序数组,反过来在切后序数组。⼀层⼀层切下去,每次后序数组最后⼀个元素就是节点元素。(每次切为两部分:中序+后序)
说到⼀层⼀层切割,就应该想到了递归
。
步骤:
第⼀步:如果后序数组⼤⼩为零的话,说明是空节点了。
第⼆步:如果不为空,那么取后序数组最后⼀个元素(某棵树的根)作为节点元素。
第三步:找到后序数组最后⼀个元素在中序数组的位置,作为切割点
第四步:切割中序数组,切成中序左数组和中序右数组 (顺序别搞反了,⼀定是先切中序数组)
第五步:切割后序数组,切成后序左数组和后序右数组
第六步:递归处理左区间和右区间
类似题目,105从前序与中序遍历序列构造二叉树