题目
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路
根据前序序列pre第一个元素找每个子树的根节点root,new一个结点并赋值,再删除pre第一个元素。
根据中序序列vin:root左右两边的的所有结点分别是root的左右子树。(分治)
显然使用递归,且需要记录vin中每个子树的起始位置。
代码
/**
* 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) {
TreeNode *p;
p=rec(pre,vin,0,vin.size()-1);
return p;
}
TreeNode* rec(vector<int> &pre,vector<int> vin,int sta,int en){
if(pre.size() && sta<=en){
TreeNode* p=new TreeNode(pre[0]);
int i=sta;
while(vin[i]!=pre[0])
i++;
pre.erase(pre.begin());
p->left=rec(pre,vin,sta,i-1);
p->right=rec(pre,vin,i+1,en);
return p;
}
return NULL;
}
};