105. 从前序与中序遍历序列构造二叉树_面试题07. 重建二叉树

问题

根据一棵树的前序遍历与中序遍历构造二叉树。

注意:
你可以假设树中没有重复的元素。

例子

思路
前序遍历：遍历顺序为 父节点 -> 左子节点 -> 右子节点
后续遍历：遍历顺序为 左子节点 -> 父节点 -> 右子节点

• 方法1
  $$

  $$

  前序遍历的第一个节点是根节点，找到根节点在中序遍历中的位置，中序遍历中，根节点之前的节点都位于左子树，根节点之后的节点都位于右子树，在前序遍历中，左子树和右子树也是分别连续的，个数同中序遍历的一样。递归求左子树和右子树
  

• 方法2
  $$

  $$

代码

//方法1
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return build(preorder, 0, preorder.length-1, inorder, 0, inorder.length-1);
    }
    
    public TreeNode build(int[] arr1, int i1, int j1, int[] arr2, int i2, int j2) {
        //j1-i1==j2-i2
        if(i1>j1) return null;
        if(i1==j1) return new TreeNode(arr1[i1]);
        
        TreeNode root = new TreeNode(arr1[i1]);
        //k为在中序遍历中根节点的位置
        int k=i2;
        for(k=i2; k<=j2; k++)
            if(arr2[k]==root.val) break;
        
        root.left = build(arr1,i1+1,i1+k-i2,arr2,i2,k-1);
        root.right = build(arr1,i1+k-i2+1,j1,arr2,k+1,j2);
        return root;
        
    }
}
//方法2