【剑指offer】重建二叉树

2018-08-10  本文已影响2人  里里角
class Solution {
public:
    TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
        if(pre.size() !=vin.size()|| vin.size()==0 ||pre.size()==0)
            return NULL;
        int len=pre.size();
        
        vector <int> left_pre,right_pre,left_vin,right_vin;
        
        TreeNode *node=new TreeNode(pre[0]);
        node->left=node->right=NULL;
        
        int i=0;
        for(;i<len;i++){
                if(vin[i]==pre[0])break;
        }
        
        vector<int> pre1(pre.begin()+1,pre.begin()+i+1); //此处i没有+1
        vector<int> vin1(vin.begin(),vin.begin()+i);
        vector<int> pre2(pre.begin()+i+1,pre.end());
        vector<int> vin2(vin.begin()+i+1,vin.end());
        node->left=reConstructBinaryTree(pre1,vin1);
        node->right=reConstructBinaryTree(pre2,vin2);
        return node;
        
        
};
}
上一篇 下一篇

猜你喜欢

热点阅读