输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
/**
* 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();
vector<int> left_pre,right_pre,left_in,right_in;
if(vinlength==0)
{
return NULL;
}
TreeNode* node=new TreeNode(pre[0]);//建立一个新的节点为根节点。
int gen=0;
for(int i=0;i<vinlength;i++)
{
if(vin[i]==pre[0])
{ gen=i;
break;}
}//记录根节点位置
for(int i=0;i<gen;i++)
{
left_in.push_back(vin[i]);//将根节点左侧中序序列压入容器
left_pre.push_back(pre[i+1]);//将除了根节点的先序遍历压入容器
}
for(int i=gen+1;i<vinlength;i++)
{
right_in.push_back(vin[i]);//根节点右侧中序压入容器
right_pre.push_back(pre[i]);//根节点右侧先序压入容器
}//至此有了四个容器分别存放根节点左右先序和中序遍历的节点信息
//利用递归,调用的函数每次返回根节点。
node->left=reConstructBinaryTree(left_pre,left_in);
node->right=reConstructBinaryTree(right_pre,right_in);
return node;
}
};
总结:题目稍难,首先搞懂先序遍历与中序遍历的联系,先序遍历的首元素是根节点,在中序遍历中找到根节点位置,左侧为左子树,右侧为右子树。先将根结点找到后建立二叉树的节点,将根结点左侧右侧的先序中序都压到容器里面,递归调用自身,形成二叉树。