常考数据结构与算法:重建二叉树

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

 

示例1

  输入

    [1,2,3,4,5,6,7],[3,2,4,1,6,5,7]

返回值

  {1,2,5,3,4,6,7}

class TestTreeNode2 {

    static TreeNode findBinaryTree(int[] pre, int preStart, int preEnd, int[] in ,int inStart, int inEnd){
        if(preStart > preEnd || inStart > inEnd)
            return null;

        TreeNode root = new TreeNode(pre[preStart]);
        for(int i = inStart; i<=inEnd; i++){
            if(in[i] == pre[preStart]){
                // 左子树的长度为i-inStart
                root.left = findBinaryTree(pre, preStart+1,preStart+i-inStart, in, inStart,i-1);
                root.right = findBinaryTree(pre, preStart+i-inStart+1,preEnd, in, i+1,inEnd);
            }
        }
        return root;
    }

    // 重建主函数
    public static TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode root = findBinaryTree(pre, 0, pre.length-1, in, 0, in.length-1);
        return root;
    }
    public static void main(String[] args) {
        int[] pre = new int[]{1,2,4,3,5,6};  // 前序序列 前序序列的第一个为根节点
        int[] in = new int[]{4,2,1,5,3,6};  // 中序序列  根节点的左边为左子树,右边为右子数

        TreeNode root = reConstructBinaryTree(pre,in);
        showPreTree(root);
    }

    static void showPreTree(TreeNode node){
        if(null == node){
            return ;
        }

        System.out.println(node.val);

        if(null != node.left){
            showPreTree(node.left);
        }

        if(null != node.right){
            showPreTree(node.right);
        }
    }
}