题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
1.根据前序序列确定根节点
2.查找中序序列中根节点的位置,在根节点之前的是左子树,后面的是右子树
3.把左子树的前序序列和中序序列,右子树的前序序列和中序序列分别重复1和2
/**
* 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* BuildNode(vector<int> pre,int pbegin,int plength,vector<int> vin,int vbegin,int vlength){
if(pbegin>=plength||vbegin>=vlength) return NULL;
TreeNode* root=new TreeNode(pre[pbegin]);
int i = 0;
for(i=vbegin;i<vlength;i++){
if(pre[pbegin]==vin[i]) break;
}
root->left=BuildNode(pre,pbegin+1,pbegin+i-vbegin+1,vin,vbegin,i);
root->right=BuildNode(pre,pbegin+i-vbegin+1,plength,vin,i+1,vlength);
return root;
}
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
return BuildNode(pre,0,pre.size(),vin,0,vin.size());}
};