剑指Offer题目：重建二叉树

剑指Offer题目：重建二叉树

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

---

解题思路

先序遍历顺序：根结点>左子结点->右子结点
中序遍历顺序：左子结点->根结点->右子结点
由先序遍历序列可知根结点，由中序遍历序列可知子结点相对于根结点的左右顺序，即知道哪些结点是根结点的左子树结点，哪些结点是根结点的右子树结点，这样就从中序序列中分割出了左子树序列和右子树序列，对两个子序列分别进行递归来构建树

---

完整代码

public class q4_reConstructBinaryTree_重建二叉树 {
    private int[] pre;
    private int[] in;

    public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
        if (pre == null || in == null || pre.length == 0 || in.length == 0 || pre.length != in.length)
            return null;
        this.pre = pre;
        this.in = in;
        return build(0, pre.length - 1, 0, in.length - 1);
    }

    private TreeNode build(int start_pre, int end_pre, int start_in, int end_in) {
        TreeNode root = new TreeNode(pre[start_pre]);
        //在中序遍历序列中找到根节点对应位置
        int root_in=start_in;
        while(root_in<in.length&&in[root_in]!=root.val)
            root_in++;
        if(root_in==in.length-1&&in[root_in]!=root.val)
            return root;//找不到对应元素说明序列有问题
        int leftLength=root_in-start_in;//中序遍历序列中左子序列的长度
        int leftendPreorder=start_pre+leftLength;//先序遍历序列中左子序列最后的下标处
        if (leftLength > 0) {
            //继续构建左子树
            root.left=build(start_pre+1,leftendPreorder,start_in,root_in-1);
        }
        if(leftLength<end_pre-start_pre)
        {
            //继续构建右子树
            root.right=build(leftendPreorder+1,end_pre,root_in+1,end_in);
        }
        return  root;
    }
}

更多LeetCode题目及答案解析见GitHub： https://github.com/on-the-roads/LeetCode
剑指offer题目及答案解析：https://github.com/on-the-roads/SwordToOffer