剑指offer第2版08题：重建二叉树

小渣渣的算法学习笔记：2018秋招备战


数据结构类算法总结：树


1.题目描述：

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

2.思路解析：

在二叉树前序遍历中，第一个数字总是树的根节点，中序遍历中，根结点位于序列的中间，左子树结点值位于根结点值的左边，右子树结点的值位于根结点值的右边，因此扫描中序遍历序列，才能找到根结点的值。

分别找到根结点的左子树右子树的前序遍历和中序遍历序列，可以用同样的方法分别构建左右子树，接下来使用递归完成即可

3.代码实现：

public class Solution04 {
    private static class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;
        }
    }

    public static void main(String[] args) {
        int pre[] = {1,2,4,7,3,5,6,8};
        int in[]  = {4,7,2,1,5,3,8,6};
        Solution04 s = new Solution04();
        TreeNode node = s.reConstructBinaryTree(pre,in);
        //中序遍历重建后的二叉树
        s.inOrder(node);
    }

    public void inOrder(TreeNode node){
        if(node != null){
            inOrder(node.left);
            System.out.print(node.val+" ");
            inOrder(node.right);
        }
    }

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

    public TreeNode constructBinaryTree(int []pre,int startPre,int endPre,int []in,int startIn,int endIn){
        //判断数组界限
        if((startPre - endPre) > 0 || (startIn - endIn) > 0) return null;
        //获取根结点
        TreeNode rootNode = new TreeNode(pre[startPre]);
        //获取根结点在中序遍历中的位置
        int rootorder = 0;
        for(int i = startIn;i<=endIn;i++){//扫描中序startIn--endIn.之前这里范围没有包含endIn位置所以一直报错StackOverFlow
            if(in[i] == pre[startPre]){
                rootorder = i;
            }
        }
        //获取左子树的长度
        int leftlength = rootorder - startIn;
        //构造左子树
        if(leftlength > 0){
            rootNode.left = constructBinaryTree(pre, startPre+1, startPre+leftlength, in, startIn, rootorder-1);
        }
        //构造右子树
        if((in.length - leftlength) > 0){
            rootNode.right = constructBinaryTree(pre, startPre+leftlength+1, endPre, in, rootorder+1, endIn);
        }
        return rootNode;
    }
}