剑指offer系列4---重建二叉树

【题目】

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

package com.exe1.offer;

/**
 * 输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。
 * 假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
 * 例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。
 * @author WGS
 *
 */
public class ConstructBinaryTree {
    //定义一个树
    public class TreeNode{
        int val;
        TreeNode left;
        TreeNode right;
        
        TreeNode(int n){
            val=n;
        }
    }
    //新建一个方法，返回重建的二叉
    public TreeNode returnConstructBinaryTree(int[] preTree,int[] inTree){        
        return constructCore(preTree,0,preTree.length-1,inTree,0,inTree.length-1);
    }
    
    public TreeNode constructCore(int[] preTree,int startPreOrder,int endPreOrder,int[] inTree,int startInOrder,int endInOrder ){
        //根据前序遍历序列得到根节点的值
        int rootValue=preTree[startPreOrder];
        //创建一个只有根结点的二叉树
        TreeNode rootNode=new TreeNode(rootValue);
        rootNode.left=null;
        rootNode.right=null;
        //只有一个结点，那么该结点即使根结点，直接返回
        if(startPreOrder==endPreOrder){
            if(startInOrder==endInOrder&&preTree[startPreOrder]==inTree[startInOrder]){
                return rootNode;
            }
        }
        
        //根据中序遍历,从中序遍历序列起点开始搜寻根结点位置
        int rootOfInTree=startInOrder;
        while(rootOfInTree<=endInOrder&&inTree[rootOfInTree]!=rootValue){
            rootOfInTree++;
        }
        
        //异常处理：中序序列中找不到这样的根结点
        while(rootOfInTree==endInOrder&&inTree[rootOfInTree]!=rootValue){
            return null;
        }
        
        //计算左子树的长度
        int leftSubTreeLen=rootOfInTree-startInOrder;
         //根据左子树的长度计算前序遍历结果中左子树的最后一个结点的下标
        int leftInOrderOfPreOrderEnd=startPreOrder+leftSubTreeLen;
        
        //重建左子树
        if(leftSubTreeLen>0){
            rootNode.left=constructCore(preTree,startPreOrder+1,leftInOrderOfPreOrderEnd,inTree,startInOrder,rootOfInTree-1);
        }
        //重建右子树
        if(leftSubTreeLen<endPreOrder-startPreOrder){
            rootNode.right=constructCore(preTree,leftInOrderOfPreOrderEnd-1,endPreOrder,inTree,rootOfInTree+1,endInOrder);
        }
        return rootNode;
        
    }
    
    public static void main(String [] args){
        int[] preTree={1,2,4,7,3,5,6,8};
        int[] inTree={4,7,2,1,5,3,8,6};
        TreeNode root=new ConstructBinaryTree().returnConstructBinaryTree(preTree, inTree);
        System.out.println(root);//返回二叉树，此处无法打印，该怎么遍历还不会。
    }
    
}