假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5
class Solution {
public:TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
//1.先判断二叉树是否为空
if((pre.size()==0)||(vin.size()==0))return NULL;
TreeNode* root=new TreeNode(pre[0]);
int i,j;
//2.中序遍历定位根节点的位置,设为i
for(i=0;i<vin.size()&&vin[i]!=pre[0];i++);
//3.前序遍历左子树存于pre_left,右子树存于pre_right,中序遍历同理
vector<int> pre_left,pre_right,vin_left,vin_right;int pre_index=1;
for(j=0;j<vin.size();j++)
{
if(j<i)
{
vin_left.push_back(vin[j]);
pre_left.push_back(pre[pre_index]);
pre_index++;
}
else if(j>i)
{
vin_right.push_back(vin[j]);
pre_right.push_back(pre[pre_index]);
pre_index++;
}
}
//4.递归调用
root->left=reConstructBinaryTree(pre_left,vin_left);root->right=reConstructBinaryTree(pre_right,vin_right);
return root;
}
};