剑指Offer——重建二叉树（前中序） Java

题目描述

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

解题思路

二叉树，想到还是递归：重建二叉树，重建根节点的左右节点，对于左右节点来说又是重建根...........

已知前序遍历，中序遍历，来重建二叉树。

1. 前序序列的首节点是当前序列重构二叉树的根节点
2. 根据根节点来划分中序序列，分为左右两部分。
3. 根据中序序列划分，在划分前序序列，分为左右两部分的子前序序列

给定：前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}

‘1’ 是当前 前序和中序 的根节点；

根据‘1’划分中序为{4,7,2} 和 {5,3,8,6}

根据中序序列的划分，再划分前序 序列为{2,4,7}和{3,5,6,8}

那么得到的前序{2,4,7}和中序{4,7,2} 就是左子树的两个序列；前序{3,5,6,8}和中序{5,3,6,8}就是右子树的两个序列

再分别对左右子树进行递归构建。

注意点：就是下标问题，这个地方需要仔细考虑下起止和结束。

对于前序序列是由中序序列的划分数目得来的，那么左子树的前序序列 的 结束下标需要减去之前划分出去的。

这个地方笔者也论述不清，建议以 前序{3,5,6,8}和中序{5,3,6,8} 当右子树时候来思考这个地方的下标。

特别注意，起始下标不是0，而是前一轮给定的 前序序列开始 和中序序列开始。

题解

    // 二叉树结构
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) { val = x; }
    }
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if (pre != null && in!= null) { // 习惯性防止出现 null传入
            TreeNode treeNode = reConstructBinaryTree(pre, in, 0, pre.length - 1, 0, in.length - 1);
            return treeNode;
        }
        return null;
    }

    private TreeNode reConstructBinaryTree(int[] pre, int[] in, int preS, int preE, int inS, int inE) {
        if (preS > preE || inS > inE ) return null;
        TreeNode root = new TreeNode(pre[preS]);

        for (int i = inS; i <= inE; i++) {
            if (in[i] == pre[preS]){
              //  当时写的时候就考虑下标了，实际上 长度来理解更方便
              //  这个 i-inS  就是中序中 根节点的左侧结点数量
                root.left = reConstructBinaryTree(pre,in,preS+1,preS+i-inS,inS,i-1);
                root.right = reConstructBinaryTree(pre,in,preS+i+1-inS,preE,i+1,inE);
                break;
            }
        }
        return root;
    }