问题描述:输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路分析:
利用循环和递归
(1)在前序遍历序列中第一个元素一定为整个树的根节点
(2)中序遍历中与此元素相等的元素之前的元素属于整个树的左子树,之后元素属于右子树
(3)由(2)可以知道在前序遍历序列中属于左子树的元素和属于右子树的元素。
(4)根据(2)(3)求得前中序遍历序列的左子树子序列和右子树子序列。对前中序遍历的左右子树进行的相同的处理可以重构整棵树
/**
* 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 len1=pre.size();
int len2=vin.size();
if(len1==0||len2==0)
return NULL;
TreeNode *root=new TreeNode(pre[0]);
int cur=pre[0];//前序遍历的第一个节点为根节点
if(pre.size()==1)return root;
vector<int> pre_left,pre_right;//存储前序遍历的左子树和右子树
vector<int> in_left,in_right;//存储中序遍历的左子树和右子树
//第一步先找到前序遍历的第一个节点(根节点)在中序遍历当中的位置,在这个位置之前的属于中序遍历的左子树,之后属于中序遍历的右子树
int pos=0;//记录根节点在中序遍历中的位置
for(int i=0;i<vin.size();i++)
if(vin[i]==cur)
pos=i;
//把中序遍历的左子树和右子树分别放入相应的向量中
int num_left=0,num_right=0;//记录左子树和右子树节点个数
for(int i=0;i<vin.size();i++)
{
if(i<pos)
{
in_left.push_back(vin[i]);
++num_left;
}
if(i>pos)
{
in_right.push_back(vin[i]);
++num_right;
}
}
//因为整个根的左右子树节点相同,所以可以求得前序遍历中左右子树的范围
for(int i=1;i<pre.size();i++)
{
if(num_left)
{
pre_left.push_back(pre[i]);
--num_left;
}
else if(num_right)
{
pre_right.push_back(pre[i]);
}
else{}
}
//第一次取出的是根节点,然后递归调用,从中序和前序遍历的左子树重建的下一个节点为根节点的左孩子,右子树为有节点
root->left=reConstructBinaryTree(pre_left,in_left);
root->right=reConstructBinaryTree(pre_right,in_right);
return root;
}
};