트리 - 중간 순서와 앞 순서, 중간 순서와 뒤 순서로 두 갈래 트리 재구성

제목: 두 갈래 나무의 앞 순서와 중간 순서의 결과를 입력하고 이 두 갈래 나무를 재건하십시오.
입력에 중복된 숫자가 없다고 가정하십시오.
예를 들어 앞 순서 반복 시퀀스 {1,2,4,7,3,5,6,8}와 중간 순서 반복 시퀀스 {4,7,2,1,5,3,8,6}를 입력하면 두 갈래 트리를 재건하고 되돌려줍니다.
1. 앞 순서와 중간 순서로 재구성
코드는 다음과 같습니다.
/**
 * 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 [] in) {
        if(pre==null||in==null||pre.length!=in.length)
            return null;
        return  helper(pre,0,pre.length-1,in,0,in.length-1);
        
    }
    public TreeNode helper(int []pre,int prelow,int prehigh,int[] in,int inlow,int inhigh)
        {
        if(prelow>prehigh||inlow>inhigh)
            return null;
        TreeNode root=new TreeNode(pre[prelow]);
        int i=inlow;
        for(;i<inhigh&&in[i]!=root.val;i++);
        
            root.left=helper(pre,prelow+1,prelow+i-inlow,in,inlow,i-1);// ;
        
            root.right=helper(pre,prelow+i-inlow+1,prehigh,in,i+1,inhigh);
        
        return root;
    }
}

2. 후순과 중순으로 재구성
코드는 다음과 같습니다.
public class Solution {
    public TreeNode buildTree(int[] inorder, int[] postorder) {
       if(inorder == null||postorder == null||inorder.length == 0||inorder.length!=postorder.length)
           return null;
        return build_Tree(inorder,0,inorder.length-1,postorder,0,postorder.length-1);         
    }
    
    public TreeNode build_Tree(int[] inorder,int inbegin,int inend,int[] postorder,int postbegin,int postend)
        {
        if(inbegin>inend||postbegin>postend)
            return null;
        
        int root_val=postorder[postend];
        
        TreeNode root=new TreeNode(root_val);
        
        int index=inbegin;
        for(;inorder[index]!=root_val;index++)
            {   
        }
        // index ;
        root.left=build_Tree(inorder,inbegin,index-1,postorder,postbegin,postbegin+index-inbegin-1);
        
        root.right=build_Tree(inorder,index+1,inend,postorder,postbegin+index-inbegin,postend-1);
        
        return root;
        
    }
}

좋은 웹페이지 즐겨찾기