牛客网 剑指Offer编程题------重建二叉树

 

题目描述 ：

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

题目链接

 首先我们要了解什么是前序遍历，什么是中序遍历。

前序遍历(Pre-order)，按照根左右的顺序沿一定路径经过路径上所有的结点。在二叉树中，先根后左再右。巧记：根左右。

中序遍历（LDR），在二叉树中，中序遍历首先遍历左子树，然后访问根结点，最后遍历右子树。巧记：左根右。

所以我们可以先在前序遍历取第一个节点，因为是前序遍历，所以这个节点必是根节点，然后我们找到它在中序遍历中的位置，

那么在中序遍历中它左边的就都是它左子树的节点了，右边就都是右子树的节点

如题，找到“1”在中序遍历的位置后，剩下的就可以利用递归延伸为找剩下子树的根和左子树与右子树。

public class Solution {

    //树节点的类
    public static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

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

    public static TreeNode reConstructBinaryTree(int[] pre, int[] in) {
        TreeNode treeNode = reBuildTheTree(pre, in, 0, pre.length - 1, 0, in.length - 1);
        return treeNode;
    }

    /**
     *  把其划分为子问题后，也需要把中序遍历与前序遍历划分为子问题对应的遍历，
     *  这里通过记录前序/中序遍历的开始与结束位置来实现。
     *
     *
     * @param pre           前序遍历的数组
     * @param in            中序遍历的数组
     * @param preStart      前序遍历的开始位置
     * @param preEnd        前序遍历的结束位置
     * @param inStart       中序遍历的开始位置
     * @param inEnd         中序遍历的结束位置
     * @return              返回该节点的对象
     */

    private static TreeNode reBuildTheTree(int[] pre, int[] in, int preStart, int preEnd, int inStart, int inEnd) {
        //如果start标志大于了end标志，说明已经结束了
        if (preStart > preEnd || inStart > inEnd) {
            return null;
        }

        TreeNode pTmp = new TreeNode(pre[preStart]);

        for (int i = 0; i <= inEnd; i++) {
            if (pre[preStart] == in[i]) {
                //把其递归为解决子树的划分问题
                pTmp.left = reBuildTheTree(pre, in, preStart + 1, preStart + i - inStart , inStart, i - 1);
                pTmp.right = reBuildTheTree(pre, in, preStart + i - inStart + 1, preEnd, i + 1, inEnd);
                break;
            }
        }

        return pTmp;
    }



    public static void main(String[] args) {
        int[] pre = new int[]{1, 2, 4, 7, 3, 5, 6, 8};
        int[] in = new int[]{4, 7, 2, 1, 5, 3, 8, 6};
        TreeNode tree = reConstructBinaryTree(pre, in);
        ShowTree(tree);

    }

    private static void ShowTree(TreeNode tree) {
        //前序遍历出结果，便于观察是否正确
        if(tree != null) {
            System.out.print(tree.val + " ");
            ShowTree(tree.left);
            ShowTree(tree.right);
        }
    }
}

其中的 preStart，preEnd 和 inStart，inEnd 

pTmp.left = reBuildTheTree(pre, in, preStart + 1, preStart + i - inStart , inStart, i - 1);
pTmp.right = reBuildTheTree(pre, in, preStart + i - inStart + 1, preEnd, i + 1, inEnd);
// i - inStart 为子树节点个数
// preStart + i - inStart 即是 preStart 开始位置加上节点个数，即是其结束位置
// preStart + i - inStart + 1 前序遍历右子树的开始位置即是左子树结束位置加1 

in（中序遍历）的位置 inStart 和 inEnd 的变化很容易理解，就是绕开根节点，前后进一个位置就是个子子树的中序遍历数组

而 pre（前序遍历）的就是先通过中序遍历来得到子树遍历的数组元素个数，然后再 preStart 的基础上挪动该个数就可以了