码迷,mamicode.com
首页 > 其他好文 > 详细

树的重构

时间:2018-11-28 22:11:28      阅读:190      评论:0      收藏:0      [点我收藏+]

标签:数组   中序   pre   turn   cat   前序遍历   注释   oca   数字   

题目:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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;
        
    }
}

树的重构

标签:数组   中序   pre   turn   cat   前序遍历   注释   oca   数字   

原文地址:https://www.cnblogs.com/c-lover/p/10034523.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!