【剑指offer】面试题7：重建二叉树（JAVA）

package sword;

// 题目：输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输
// 入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,
// 2, 4, 7, 3, 5, 6, 8}和中序遍历序列{4, 7, 2, 1, 5, 3, 8, 6}，则重建出
// 二叉树并输出它的头结点。

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

    TreeNode(int x) {
        val = x;
    }
}

class TreeClass{

    public TreeNode buildTree(int[] preorder, int[] inorder) {
        return construct(preorder, inorder, 0, preorder.length - 1, 0, inorder.length - 1);
    }
    public TreeNode construct(int[] preorder, int[] inorder,int pre_start,int pre_end,int in_start,int in_end){
        if(pre_start>pre_end||in_start>in_end) return null;
        //在中序里找根结点
        int i = in_start;
        while(inorder[i] != preorder[pre_start])    i++;
        TreeNode b = new TreeNode(preorder[pre_start]);
        //根据根结点在中序中的位置，划分为左子树和右子树，迭代。重点是找对左右子树前序的左右边界。
        b.left = construct(preorder, inorder,pre_start+1,pre_start+i-in_start,in_start,i-1);
        b.right = construct(preorder, inorder,pre_end-in_end+i+1,pre_end,i+1,in_end);
        return b;
    }
}

/**
 * @author PangWanjia
 * @date 2021/3/30 21:01
 */
public class test07 {
    public static void main(String[] args) {

    }
}