一、题目
解法一—— 递归
class Solution {
public:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin)
{
if(pre.size() == 0) return nullptr;
TreeNode* root = new TreeNode(pre[0]);
int temp;
root->val = pre[0];
temp = index(pre[0],vin);
vector<int> pre1,pre2,vin1,vin2;
for(int i=1;i<temp+1;i++)
{
pre1.push_back(pre[i]);
}
for(int i=0;i<temp;i++)
{
vin1.push_back(vin[i]);
}
for(int i=temp+1;i<vin.size();i++)
{
pre2.push_back(pre[i]);
}
for(int i=temp+1;i<vin.size();i++)
{
vin2.push_back(vin[i]);
}
root->left = reConstructBinaryTree(pre1,vin1);
root->right = reConstructBinaryTree(pre2,vin2);
return root;
}
int index(int a,vector<int> b)
{
int ans = 0;
if(b.size() == 0) return 0;
while(a!=b[ans])
{
ans++;
}
return ans;
}
};
对于中序遍历和前序遍历,前序0中可获得根节点数值,然后通过查找中序中与前序根节点相同大小的节点可获得左树和右树的vector,然后进行递归。
push_back() vector最后放入元素。