Leetcode:Construct Binary Tree 앞 순서와 중간 순서, 뒤 순서와 중간 순서로 두 갈래 트리 만들기

8350 단어 LeetCode
앞 순서와 중간 순서로 두 갈래 트리 만들기
후순과 중순으로 두 갈래 트리 구축
분석: 주요 사고방식은 중서에서 뿌리 노드를 찾아 좌우 나무를 구분하는 것이다. 구체적으로는 다음과 같다.
1. 루트 노드를 찾습니다.우리는 전차 서열의 첫 번째 원소와 후차 서열의 마지막 원소가 틀림없이 루트 노드라는 것을 안다. 우리는 이것을 돌파구로 삼는다
2. 루트 노드의 좌표를 결정합니다.우리는 중간 서열에서 루트 노드의 아래 표를 찾았다.
3. 좌우 나무를 나눈다.
중렬 시퀀스에 대해: 루트 노드가 아래에 표시되기 전의 것은 모두 왼쪽 트리에 속하고, 그 후의 것은 모두 오른쪽 트리에 속한다
선행 시퀀스의 경우: 루트 노드 이후의 요소 (이 세그먼트 길이는 중간 시퀀스의 왼쪽 트리 길이에 따라 결정) 는 왼쪽 트리에 속하고, 왼쪽 트리 이후의 요소는 오른쪽 트리에 속합니다.
선행 시퀀스의 경우: 루트 노드 이전의 요소 (이 세그먼트 길이는 중간 시퀀스의 오른쪽 트리 길이에 의해 결정될 수 있음) 는 오른쪽 트리에 속하고, 오른쪽 트리 이후의 요소는 왼쪽 트리에 속합니다.
4. 분할된 좌우 트리에 대해 두 갈래 트리를 구축하고 각 단락의 서열의 아래 표시 범위를 잘 확정한다.
예:
preorder = {7,10,4,3,1,2,8,11}
inorder = {4,10,3,1,7,11,8,2}

첫 번째 단계: 루트 노드를 찾습니다. 순서에 따라 루트 노드는 첫 번째 요소, 즉 7입니다.
두 번째 단계: 루트 노드의 좌표를 확인하고, 우리는 중간 순서에서 루트 노드 7을 찾고, 아래에 4라고 표시하는 것을 확인한다
3단계: 좌우 나무를 분할한다.
중렬 서열에 대해: 왼쪽 트리는 {4, 10, 3, 1}, 오른쪽 트리는 {11, 8, 2}
선순 시퀀스에 대해: 왼쪽 트리는 {10, 4, 3, 1}, 오른쪽 트리는 {2, 8, 11}
주의: 언제든지 중차 서열과 선차 서열의 길이는 같다.
class Solution {
public:
    TreeNode *buildTree(vector<int> &preorder, vector<int> &inorder) {
        return constructTree(preorder, 0, preorder.size() - 1, inorder, 0, inorder.size() - 1);
    }
    
    TreeNode* constructTree(const vector<int>& preorder, int preFirst, int preLast,
                            const vector<int>& inorder, int inFirst, int inLast) {
        int preLen = preLast - preFirst + 1;
        int inLen  = inLast  - inFirst  + 1;
        assert(preLen == inLen && preLen >= 0);
        if (preLen == 0) return nullptr;
        
        TreeNode* root = new TreeNode(preorder.at(preFirst));
        int rootIndex = distance(inorder.begin(), find(inorder.begin(), inorder.end(), preorder.at(preFirst)));
        int leftLen  = rootIndex - inFirst;
        int rightLen = inLast - rootIndex;
        root->left =  constructTree(preorder, preFirst + 1, preFirst + leftLen, inorder, inFirst, rootIndex - 1);
        root->right = constructTree(preorder, preLast - rightLen + 1, preLast, inorder, rootIndex + 1, inLast);
        return root;
    }
};

 
class Solution {
public:
    TreeNode *buildTree(vector<int> &inorder, vector<int> &postorder) {
        return constructTree(inorder, 0, inorder.size() - 1, postorder, 0, postorder.size() - 1);
    }
     TreeNode* constructTree(const vector<int>& inorder, int inFirst, int inLast,
                             const vector<int>& postorder, int postFirst, int postLast) {
        int postLen = postLast - postFirst + 1;
        int inLen  = inLast  - inFirst  + 1;
        assert(postLen == inLen && postLen >= 0);
        if (postLen == 0) return nullptr;
        
        TreeNode* root = new TreeNode(postorder.at(postLast));
        int rootIndex = distance(inorder.begin(), find(inorder.begin(), inorder.end(), postorder.at(postLast)));
        int leftLen  = rootIndex - inFirst;
        int rightLen = inLast - rootIndex;
        root->left =  constructTree(inorder, inFirst, rootIndex - 1, postorder, postFirst, postFirst + leftLen - 1);
        root->right = constructTree(inorder, rootIndex + 1, inLast, postorder, postLast - rightLen, postLast - 1);
        return root;
    }
};

좋은 웹페이지 즐겨찾기