剑指offer JZ7 重建二叉树 Java实现

1. 题目描述

2.思路分析

这道题在陈越姥姥的数据结构课上做过一道类似的题目TreeTraveralsAgain，那道题目是根据前序遍历的顺序和中序遍历的顺序来输出后续遍历，当时是看着姥姥的视频，采用分治法的思想进行解决，没有仔细考虑常规解法去重新生成一棵二叉树，再后序遍历。那么我想是不是那个的算法稍作改动可以解决这道题目呢？

没错，和那个题思路基本一毛一样，Divide and Conquer。根据前序遍历序列pre的第一个值确定根在中序遍历序列vin中的位置，然后把树的问题转换成递归求解更小规模的左右两棵子树的问题。然后把这个函数包装一下成为解题的方法。要注意的问题：

1.非法输入与空数组都要返回null

2.左右子树在pre和vin中下标的问题：

对于一棵树有：先序序列最左端下标=preL，中序序列最左端下标=inL, 中序序列最右端下标=inR，找到pre[preL]在vin[]中的下标为position
左子树：先序序列最左端下标=preL+1，中序序列最左端下标=inL,中序序列最右端下标=position-1.
右子树：先序序列最左端下标=preL+（左子树的结点数）+1=preL+(position-inL)+1，中序序列最左端下标=postion+1,中序序列最右端下标=inR.

3.实现代码

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] vin) {
        int preL=0,inL=0,inR=pre.length-1;//指向前序中序两个数组最左端的下标
        return getAndBuildRoot(preL,inL,inR,pre,vin);
    }
    private TreeNode getAndBuildRoot(int preL,int inL,int inR,int [] pre,int [] vin){
        //特殊情况：1.传进来是空的数组 2.只有一个结点 3.左右指针已经交错
        if(inL>inR||preL>pre.length-1||pre.length==0)
            return null;
        //建立一个根节点
        TreeNode root= new TreeNode(pre[preL]);
        //记录当前根节点在in数组中的位置
        int position=0;
        //找到这个根结点在vin中的位置
        for(int i=inL;i<=inR;i++){
            if(vin[i]==pre[preL]){
                position=i;
                break;
            }        
        }
        root.left=getAndBuildRoot(preL+1,inL,position-1,pre,vin);
        root.right=getAndBuildRoot(preL+(position-inL)+1,position+1,inR,pre,vin);
        return root;
    }