lc_rt_105_buildTree

题目：从前序与中序遍历序列构造二叉树 middle

根据一棵树的前序遍历与中序遍历构造二叉树。
注意:
你可以假设树中没有重复的元素。

例如，给出
前序遍历 preorder = [3,9,20,15,7]
中序遍历 inorder = [9,3,15,20,7]

返回如下的二叉树：

    3
   / \
  9  20
    /  \
   15   7

 

 

 

package leetCode.Recursion_Tacktrace;

import java.util.Arrays;
import java.util.Map;

public class lc_rt_105_buildTree {

/*
思路：
本题和剑指offer4一样，虽然做出来，但是花了很久

对于任意一颗树而言，前序遍历的形式总是：
[ 根节点, [左子树的前序遍历结果], [右子树的前序遍历结果] ]

即根节点总是前序遍历中的第一个节点。而中序遍历的形式总是：
[ [左子树的中序遍历结果], 根节点, [右子树的中序遍历结果] ]

    只要我们在中序遍历中定位到根节点，那么我们就可以分别知道左子树和右子树中的节点数目。
由于同一颗子树的前序遍历和中序遍历的长度显然是相同的，因此我们就可以对应到前序遍历的结果中，
对上述形式中的所有左右括号进行定位。
    这样以来，我们就知道了左子树的前序遍历和中序遍历结果，以及右子树的前序遍历和中序遍历结果，
我们就可以递归地对构造出左子树和右子树，再将这两颗子树接到根节点的左右位置。



*/

    public static void main(String[] args) {
        lc_rt_105_buildTree m = new lc_rt_105_buildTree();
        int[] preorder = {1, 2, 4, 5, 3, 6, 8, 9, 7};
        int[] inorder = {4, 2, 5, 1, 8, 6, 9, 3, 7};
        TreeNode root = m.buildTree(preorder, inorder);
        root.preOrder();
    }



    public TreeNode buildTree(int[] preorder, int[] inorder) {
        // 数组长度为0时，要处理
        if (inorder.length == 0)
            return null;
        // 1.使用递归要确定返回点，数组长度为1时，即为根结点，返回子树的根结点
        if (inorder.length == 1)
            return new TreeNode(inorder[0]);

        // 2.找到root在中序遍历时的下标rootIndex
        int rootIndex = -1;
        for (int j = 0; j < inorder.length; j++) {
            if (inorder[j] == preorder[0]) {
                rootIndex = j;
                break;
            }
        }
        // 3.递归。
        // 中序遍历序列中，rootIndex左边为左子树的中序遍历序列，rootIndex右边为右子树的中序遍历序列
        // 前序遍历序列中:
        // [1,1 + leftInorder.length-1]为左子树的前序遍历序列（和leftinorder个数相同），
        // [1 + leftInorder.length,preorder.length-1]为右子树的前序遍历序列

        // 数学归纳法，假设最后一步，root左右子树已建立好，只要将左右子树的根节点安上去即可。

        // 3.1 找到左子树根结点。Arrays.copyOfRange(int[] arr,int start,int end),有头无尾
        // 3.2 找到右子树根结点。Arrays.copyOfRange(int[] arr,int start,int end),有头无尾
        int[] leftInorder = Arrays.copyOfRange(inorder, 0, rootIndex);
        int[] rightInorder = Arrays.copyOfRange(inorder, rootIndex + 1, inorder.length);
        int[] leftPreOrder = Arrays.copyOfRange(preorder, 1, 1 + leftInorder.length);
        int[] rightPreOrder = Arrays.copyOfRange(preorder, 1 + leftInorder.length, preorder.length);

        // 3.3 安上左右结点
        TreeNode root = new TreeNode(preorder[0]);
        root.left = buildTree(leftPreOrder, leftInorder);
        root.right = buildTree(rightPreOrder, rightInorder);

        return root;

    }

    class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }

        public void preOrder() {
            System.out.println(this.val);
            if (this.left != null)
                this.left.preOrder();
            if (this.right != null)
                this.right.preOrder();
        }
    }

}