树的重构

Posted 2021-01-23 c-lover

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

思路：见注释

public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        TreeNode node=ConstructBinary(pre, 0, pre.length-1, in, 0, in.length-1);
        return node;
        
    }
    
  
    /**
    *pre 前序遍历的数组
    *ps  前序遍历的开始位置
    *pe  前序遍历的结束位置
    *in  中序遍历的数组
    *is  中序遍历的开始位置
    *ie  中序遍历的结束位置
    */
    public TreeNode ConstructBinary(int[] pre,int ps,int pe,int[] in,int is,int ie){
        //前序遍历完返回叶子节点
        if(pe<ps){
            return null;
        }
        int value=pre[ps];
        //int index=location(in,value);//找到根节点在中序遍历中的位置
        int index=is;
        while(index<in.length&&in[index]!=value){
            index++;
        }
        TreeNode node=new TreeNode(value);
        //node左子树的个数为index-is
        //所以对于左子树来说前序的ps=ps+1 pe=ps+index-is  中序的is=is   ie=index-1
        node.left=ConstructBinary(pre, ps+1, ps+index-is, in, is, index-1);
        
        //node右子树的个数为ie-index
        //所以右子树的前序ps=pe-ie+index pe=pe 中序的is=index+1,ie=ie
        node.right=ConstructBinary(pre, ps+index-is+1, pe, in, index+1, ie);
        return node;
        
    }
}