4. 두 갈래 나무 재건

2172 단어
제목은 어떤 두 갈래 나무의 앞 순서와 중간 순서를 입력한 결과를 설명합니다. 이 두 갈래 나무를 다시 만드십시오.입력한 앞 순서와 중간 순서의 결과에 중복된 숫자가 없다고 가정하십시오.예를 들어 앞 순서 반복 시퀀스 {1,2,4,7,3,5,6,8}와 중간 순서 반복 시퀀스 {4,7,2,1,5,3,8,6}를 입력하면 두 갈래 트리를 재건하고 되돌려줍니다.
/**
 * 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) {
        TreeNode root = new TreeNode(pre[0]);
        int i = 0;
        while(in[i]!=root.val){
            i++;
        }
        if(i==0){
            root.left = null;
        }else{
            int [] leftPre = new int[i];
            System.arraycopy(pre,1,leftPre,0,i);
            int [] leftIn = new int[i];
            System.arraycopy(in,0,leftIn,0,i);
            root.left = reConstructBinaryTree(leftPre,leftIn);
            
        }
        
        if(i==in.length-1){
            root.right=null;
        }else{
            int [] rightPre = new int[pre.length-i-1];
            System.arraycopy(pre,i+1,rightPre,0,pre.length-i-1);
            int [] rightIn = new int[pre.length-i-1];
            System.arraycopy(in,i+1,rightIn,0,pre.length-i-1);
            root.right = reConstructBinaryTree(rightPre,rightIn);
        }

        return root;
    }
}

두 번째 할 때 쓴 코드:
/**
 * 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.length!=in.length||pre.length==0){
            return null;
        }
        
        TreeNode root = new TreeNode(pre[0]);
        
        
        // 
        int rootIndex = -1;
        for(int i=0; i

좋은 웹페이지 즐겨찾기