输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
先序遍历第一个位置肯定是根节点node,
中序遍历的根节点位置在中间p,在p左边的肯定是node的左子树的中序数组,p右边的肯定是node的右子树的中序数组
另一方面,先序遍历的第二个位置到p,也是node左子树的先序子数组,剩下p右边的就是node的右子树的先序子数组
把四个数组找出来,分左右递归调用即可.
代码:
/**
* 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 in_size = vin.size();
if(in_size == 0) //当中序遍历数组为空时,表示无相应子树,返回子树为空
return NULL;
int p = pre[0];
TreeNode * node = new TreeNode(p); //利用先序遍历数组第一个值作为二叉树的头结点
vector<int>pre_left,pre_right,vin_left,vin_right;
int j=0;
for(int i = 0;i< vin.size();i++){
if(vin[i]==p){
j=i; //记录头结点的值在中序遍历数组中的索引
break; //剪枝操作
}
}
for(int k=0;k<vin.size();k++){ //把中序遍历数组中头结点左边的数存到vin_left
if(k<j){
vin_left.push_back(vin[k]);
}
else if(k>j)
{
vin_right.push_back(vin[k]); //把中序遍历数组中头结点右边的数存到vin_right
}
}
for(int q=1;q<pre.size();q++){
if(q<=j)
pre_left.push_back(pre[q]); //把先序遍历数组中头结点的左子树的值存到
//pre_left
else
pre_right.push_back(pre[q]); //把先序遍历数组中头结点的右子树的值存到
//pre_right
}
node->left = reConstructBinaryTree(pre_left,vin_left);//重建头结点的左子树,并返回
//该左子树的的头结点作为父亲结点
//的左孩子
node->right =reConstructBinaryTree(pre_right,vin_right);//重建头结点的右子树,并返回
//该右子树的的头结点作为父亲结点
//的右孩子
return node; //返回父亲结点
}
};