题目
根据一棵树的前序遍历与中序遍历构造二叉树。
注意:你可以假设树中没有重复的元素。
例如,给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]
返回如下的二叉树:
前序:根左右,先处理根节点,再处理左子树,最后右子树
中序:确定根节点的位置,划分出左右子树的下标范围
HashMap<Integer, Integer> map = new HashMap<>();
int index = 0;//从前面开始,依次找各棵树的根节点
public TreeNode buildTree(int[] preorder, int[] inorder) {
// 前序:根左右,根在前面
// 中序:左根右
// 先找根
for (int i = 0; i < inorder.length; i++)
map.put(inorder[i], i);
return helper(preorder, inorder, 0, inorder.length - 1);
}
public TreeNode helper(int[] preorder, int[] inorder, int left, int right) {
if (left > right) {
return null;
}
int val = preorder[index++];
TreeNode root = new TreeNode(val);
root.left = helper(preorder, inorder, left, map.get(val) - 1);
root.right = helper(preorder, inorder, map.get(val) + 1, right);
return root;
}
VS
题目
根据一棵树的中序遍历与后序遍历构造二叉树。
注意:你可以假设树中没有重复的元素。
例如,给出
中序遍历 inorder = [9,3,15,20,7]
后序遍历 postorder = [9,15,7,20,3]
1.HashMap存中序遍历的元素与下标的关系
2.通过后序遍历找根节点的的值,再在中序遍历中找到此根节点的下标,从而划分左右子树
(后序遍历左右根,最后一个是根节点,所以设置一个变量从最后一个开始往前指)
3.递归创建数,先创建树根,给树根赋值,然后再分别讨论左右子树
4.如何创建左右子树:通过根节点的下标划分左右子树的范围,递归下去,范围逐渐缩小
HashMap<Integer, Integer> map = new HashMap<>();
int index;// 设置一个变量从最后一个开始往前指,以找到以该节点为根的子树
public TreeNode buildTree(int[] inorder, int[] postorder) {
for (int i = 0; i < inorder.length; i++)
map.put(inorder[i], i);
index = postorder.length - 1;
return helper(inorder, postorder, 0, inorder.length - 1);
}
public TreeNode helper(int[] inorder, int[] postorder, int left, int right) {
if (left > right) {
return null;
}
int val = postorder[index--];
TreeNode root = new TreeNode(val);
root.right = helper(inorder, postorder, map.get(val) + 1, right);// 后序变量左右根,倒着来,先根,然后右子树,最后左子树
root.left = helper(inorder, postorder, left, map.get(val) - 1);
return root;
}
总结
前序+中序:依据前序的根左右顺序来创建左右子树(先左后右)
后序+中序:依据后序的左右根来创建左右子树(先右后左)
都是通过中序遍历来划分左右子树的下标范围。