889. 전차와 후차에 따라 두 갈래 나무를 반복적으로 구성한다.

2308 단어
주어진 전차와 후차와 일치하는 두 갈래 트리를 되돌려줍니다.pre와post가 두루 다니는 값은 서로 다른 정수입니다.예: 입력: pre = [1,2,4,5,3,6,7],post = [4,5,2,6,7,3,1] 출력: [1,2,3,4,5,6,7] 알림: 1 <= pre.length == post.length<=30 pre[]와post[]는 모두 1, 2,......,pre.length의 배열은 입력마다 적어도 하나의 답이 있어야 한다.만약 여러 개의 답이 있다면, 그 중 하나로 돌아갈 수 있다.
생각
왼쪽 지점에 LL 노드를 지정합니다.우리는 왼쪽 지점의 두 노드가pre[1]라는 것을 알고 있지만, 그것도 왼쪽 지점의 뒷순서가 표시하는 마지막에 나타난다.그래서 pre[1]=post[L-1](결점의 값이 유일하기 때문에), L=post.indexOf(pre[1]) + 1. 현재 우리의 귀속 단계에서 왼쪽 지점은pre[1:L+1]과post[0:L]에서 다시 분할되고, 오른쪽 지점은pre[L+1:N]과post[L:N-1]에서 다시 분할됩니다.

코드

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode constructFromPrePost(int[] pre, int[] post) {
        if (pre.length != post.length) {
            return null;
        } 
        
        return buildMyTree(pre, 0, pre.length-1, post, 0, post.length-1);
    }
    
    public TreeNode buildMyTree(int[] pre,
                                int preStart,
                                int preEnd,
                                int[] post,
                                int postStart,
                                int postEnd) {
        if (preStart == preEnd) {
            return new TreeNode(pre[preStart]);
        }
        if (preStart > preEnd) {
            return null;
        }
        if (postStart > postEnd) {
            return null;
        }
        
        int root_value = pre[preStart];
        TreeNode root = new TreeNode(root_value);
        int root_left_value = pre[preStart+1];
        int position = findPosition(post, postStart, postEnd, root_left_value);
        if (position == -1) {
            return null;
        }
        
        root.left = buildMyTree(pre, preStart+1, preStart+1+position-postStart, post, postStart, position);
        root.right = buildMyTree(pre, preStart+position-postStart+2, preEnd, post, position+1, postEnd-1);
        
        return root;
    }
    
    public int findPosition(int[] post, int start, int end, int key) {
        for (int i = start; i <= end; i++) {
            if (post[i] == key) {
                return i;
            }
        }
        return -1;
    }
}

좋은 웹페이지 즐겨찾기