题目:给定一颗树前序遍历与中序遍历。构造二叉树并返回根节点。
示例1:
3
/ \
9 20
/ \
15 7
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
思路:
本质上思路就是平时考试试卷上给你前序遍历,中序遍历,还原二叉树(不是废话,让你用笔写,你肯定会,思路就是用笔写出答案的思路)。首先根据前序遍历确定根节点,之后找到根节点在中序遍历的位置,左侧的就都是左子树,右侧的就都是右子树。确定了左子树、右子树的元素个数,使用递归完成题目。
代码:
class Solution {
private Map<Integer, Integer> helpMap;
public TreeNode makeTree(int[] preorder, int[] inorder, int pre_left, int pre_right, int in_left, int in_right) {
//边界条件
if (pre_left > pre_right||in_left>in_right) {
return null;
}
// 前序遍历中的第一个节点是根节点
int pre_root = pre_left;
// 在中序遍历中找到根节点
int in_root = helpMap.get(preorder[pre_root]);
// 先把根节点建立出来
TreeNode root = new TreeNode(preorder[pre_root]);
// 得到左子树中的节点数目
int left_sum = in_root - in_left;
// 递归地构造左子树,并连接到根节点
// 先序遍历中从左边界+1 开始的 left_sum个元素就对应了中序遍历中从左边界到根节点-1的元素
root.left = makeTree(preorder, inorder, pre_left + 1, pre_left + left_sum, in_left, in_root - 1);
// 递归地构造右子树,并连接到根节点
// 先序遍历中从左边界+左子树节点数目+1开始到右边界的元素就对应了中序遍历中从根节点+1到右边界的元素
root.right = makeTree(preorder, inorder, pre_left + left_sum + 1, pre_right, in_root + 1, in_right);
return root;
}
public TreeNode buildTree(int[] preorder, int[] inorder) {
int n = preorder.length;
helpMap = new HashMap<Integer, Integer>();
for (int i = 0; i < n; i++) {
//通过值来找索引,所以值为键
helpMap.put(inorder[i], i);
}
return makeTree(preorder, inorder, 0, n - 1, 0, n - 1);
}
}