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;
};
}
网友评论