검지offer07_두 갈래 나무를 재건하다

제목: 두 갈래 나무의 앞 순서와 중간 순서의 결과를 입력하십시오. 이 두 갈래 나무를 다시 만드십시오. (중복 숫자가 없는 경우)트리 노드는 다음과 같이 정의됩니다.

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) :val(x), left(nullptr), right(nullptr) {}
};

문제 풀이 사고방식: 1.앞의 첫 번째 숫자는 항상 나무의 루트 노드의 값입니다.그러나 중차 반복에서 루트 노드의 값은 시퀀스 중간에 있고 왼쪽 트리의 노드의 값은 루트 노드의 값 왼쪽에 있고 오른쪽 트리의 노드의 값은 루트 노드의 값 오른쪽에 있습니다.루트 노드의 값을 찾을 수 있도록 시퀀스를 스캔합니다.2. 또한 앞의 순서가 뿌리 노드 뒤에 왼쪽 트리를 먼저 훑어봐야 오른쪽 트리를 훑어볼 수 있기 때문에 뒤의 순서가 왼쪽 트리의 노드 수를 추정한 후에 앞의 루트 노드를 훑어본 후의 같은 수의 값은 왼쪽 트리의 노드 값이다.나머지는 오른쪽 나무의 값이다.이렇게 해서 좌우 트리 서열을 각각 찾았다.3. 뿌리 노드와 좌우자 나무의 전순, 중순을 각각 확정했다. 우리는 같은 방법으로 좌우자 나무를 구축할 수 있고 나머지는 귀속으로 완성할 수 있다.
Solution:

#include 
#include 
#include 
#include 

using namespace std;

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) :val(x), left(nullptr), right(nullptr) {}
};

class Solution{
public:
    TreeNode* reConstructBinaryTree(vector pre, vector vin){
        if ( pre.size() == NULL || pre.size() != vin.size())
            return NULL;

        int root = pre[0];
        TreeNode *newNode = new TreeNode(root);
        if(pre.size() == 1)
            return newNode;

        int vinRootPost = 0;
        for(int i = 0; i < vin.size(); i++){
            if (vin[i] == root )
                vinRootPost = i;
        }

        vector a;
        for(int i = 1; i < vinRootPost+1;i++){
            a.push_back(pre[i]);
        }
        vector b;
        for (int j = vinRootPost+1; j < pre.size(); j++) {
            b.push_back(pre[j]);
        }
        vector c;
        for(int k = 0; k < vinRootPost;k++){
            c.push_back(vin[k]);
        }
        vector d;
        for (int z = vinRootPost+1; z < vin.size(); z++) {
            d.push_back(vin[z]);
        }

        newNode->right = reConstructBinaryTree(b,d);
        newNode->left = reConstructBinaryTree(a,c);

        return newNode;

    }

    void postOrder(TreeNode *root){
        if(root->left)
            postOrder(root->left);
        if(root->right)
            postOrder(root->right);
        printf("%d", root->val);
    }

};

int main()
{
    vector pre{1,2,4,5,3,6,7};
    vector vin{4,2,5,1,6,3,7};

    TreeNode *newRoot = Solution().reConstructBinaryTree(pre, vin);
    Solution().postOrder(newRoot);

    return 0;
}