검지 offer 프로그래밍 문제 (4): 두 갈래 나무 재구성

제목은 어떤 두 갈래 나무의 앞 순서와 중간 순서를 입력한 결과를 설명합니다. 이 두 갈래 나무를 다시 만드십시오.입력한 앞 순서와 중간 순서의 결과에 중복된 숫자가 없다고 가정하십시오.예를 들어 앞 순서 반복 시퀀스 {1,2,4,7,3,5,6,8}와 중간 순서 반복 시퀀스 {4,7,2,1,5,3,8,6}를 입력하면 두 갈래 트리를 재건하고 되돌려줍니다.
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
         if(vin.size()==0 || pre.size() == 0)
            return NULL;
        int k;
        for(int i=0;iif(pre[0]==vin[i]){
                k=i;
                break;
            }
        }
        TreeNode *node=new TreeNode(pre[0]);
        vector<int> left_vin;
        vector<int> left_pre;
        vector<int> right_vin;
        vector<int> right_pre;
        for(int i=0;i1]);
        }
        // {1,2,4,7,3,5,6,8}
        // {4,7,2,1,5,3,8,6}
        for(int i=k+1;ileft=reConstructBinaryTree(left_pre,left_vin);
        node->right=reConstructBinaryTree(right_pre,right_vin);
        return node;
    }
};
class Solution
{
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)
    {
        if(pre.size()<=0||vin.size()<=0)
            return NULL;
        TreeNode *root=new TreeNode(pre[0]);
        // {1,2,4,7,3,5,6,8} {4,7,2,1,5,3,8,6}
        for(int i=0;iif(pre[0]==vin[i])
            {

                root->left=reConstructBinaryTree(copy(pre,1,i),copy(vin,0,i-1));
                root->right=reConstructBinaryTree(copy(pre,i+1,pre.size()-1),copy(vin,i+1,vin.size()-1));                            
                break;
            }

        }
        return root;
    }
    vector<int> copy(vector<int> array,int begin,int end)
    {
        vector<int> temp;
        while(begin<=end)
        {
            temp.push_back(array[begin++]);
        }
        return temp;
    }
};

두 갈래 나무는 뿌리 노드를 제거하고 왼쪽 아이와 오른쪽 아이는 모두 새로운 나무이다. 앞의 순서는 {1,2,4,7,3,5,6,8}이고 중간의 순서는 뿌리 노드의 위치에서 분리된다. 그의 왼쪽 아이의 앞의 순서는 {2,4,7}이고 오른쪽 아이는 {3,5,6,8}로 추정된다.먼저 루트 노드가 중순으로 흐르는 위치를 찾은 다음에 좌우 트리를 분리하여 트리를 만듭니다.
교체기 방법으로 실현:
class Solution
{
public:
    TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin)
    {
        if (pre.size() <= 0 || vin.size() <= 0)
            return NULL;

        // {1,2,4,7,3,5,6,8} {4,7,2,1,5,3,8,6}
        TreeNode* node = new TreeNode(pre[0]);
        for (auto itVin = vin.begin(); itVin != vin.end(); ++itVin)
            if (*itVin == pre[0]) {
                auto itPre = pre.begin() + 1 + (itVin - vin.begin());
                node->left = reConstructBinaryTree(copy(pre,pre.begin() + 1, itPre), copy(vin,vin.begin(), itVin));
                node->right = reConstructBinaryTree(copy(pre,itPre, pre.end()), copy(vin,itVin + 1, vin.end()));
            }
        return node;
    }
    vector<int> copy(vector<int> array, vector<int> ::iterator be, vector<int> ::iterator en)
    {
        vector<int> temp;
        temp.assign(be, en);
        return temp;
    }
};
class Solution
{
public:
    TreeNode* reConstructBinaryTree(vector<int> pre, vector<int> vin)
    {
        if (pre.size() <= 0 || vin.size() <= 0)
            return NULL;

        // {1,2,4,7,3,5,6,8} {4,7,2,1,5,3,8,6}
        TreeNode* node = new TreeNode(pre[0]);
        for (auto itVin = vin.begin(); itVin != vin.end(); ++itVin)
            if (*itVin == pre[0]) {
                auto itPre = pre.begin() + 1 + (itVin - vin.begin());
                node->left = reConstructBinaryTree(vector<int>(pre.begin() + 1, itPre), vector<int>(vin.begin(), itVin));
                node->right = reConstructBinaryTree(vector<int>(itPre, pre.end()), vector<int>(itVin + 1, vin.end()));
            }
        return node;
    }

};

좋은 웹페이지 즐겨찾기