重建二叉树

2018-08-28  本文已影响5人  稀饭粥95

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

public class Solution {
        public int c = -1;
    public TreeNode construction(int s,int e,int[] pre,int []in){
        if(s<=e){
            c++;
            TreeNode node = new TreeNode(pre[c]);
            int mid = 0;
            for(int i=s;i<=e;i++){
                if(in[i]==pre[c]){
                    mid=i;
                    break;
                }
            }
            node.left=construction(s,mid-1,pre,in);
            node.right=construction(mid+1,e, pre, in);
            return node;
            
        }else{
            return null;
        }
        
        
    }
    
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length==0){
            return null;
        }
        return construction(0,pre.length-1,pre,in);
    }
}
上一篇 下一篇

猜你喜欢

热点阅读