解题思路如下:
1、首先建立4个vector用来存放前序遍历和中序遍历的左右子树;
2、由于前序遍历的第1个元素就是根节点,则在中序遍历中找到它记为Head,则中序遍历中Head左边的长度为Head的几个元素为中序遍历的左子树;同理在前序遍历中第1个元素右边长度为Head的几个元素为前序遍历的左子树;
3、然后以Head为分界线,把他们的左右子树分别放入各自新建的vector中;
4、接着对他们的4个左右子树再递归重新重建二叉子树
5、整个过程就是不断找Head,不断分为4个子树,再递归调用重建二叉树;
实现代码如下:
/**
* 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) {
int vinlength=vin.size();
if(vinlength==0)
return NULL;
TreeNode* root=new TreeNode(pre[0]);
vector<int>pre_left,pre_right,vin_left,vin_right;
int Head=0;
for(int i=0;i<vinlength;i++)
{
if(vin[i]==pre[0])
{
Head=i;
break;
}
}
for(int i=0;i<Head;i++)
{
vin_left.push_back(vin[i]);
pre_left.push_back(pre[i+1]);
}
for(int i=Head+1;i<vinlength;i++)
{
vin_right.push_back(vin[i]);
pre_right.push_back(pre[i]);
}
root->left=reConstructBinaryTree(pre_left,vin_left);
root->right=reConstructBinaryTree(pre_right,vin_right);
return root;
}
};