题目:
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
思路:
pre: 1 2 4 7 3 5 6 8
in: 4 7 2 1 5 3 8 6
每次取前序遍历的第一个值,然后在中序遍历中找到该值所在的下标,那么新建该节点为root, root->left指向中序遍历左边构成的子树,root->right指向中序遍历右边构成的子树。依次递归下去。
代码:
* 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 *root = reConstructBinaryTree(pre,0,pre.size()-1,vin,0,vin.size()-1);
return root;
}
private:
TreeNode * reConstructBinaryTree(vector<int> pre, int pre_start,int pre_end,vector<int>vin,int vin_start,int vin_end){
if(pre_start>pre_end || vin_start>vin_end){
return NULL;
}
TreeNode *root = new TreeNode(pre[pre_start]);
for(int i=vin_start;i<=vin_end;i++){
if(vin[i]==pre[pre_start]){
root->left= reConstructBinaryTree(pre,pre_start+1,i-vin_start+pre_start,vin,vin_start,i-1);
root->right= reConstructBinaryTree(pre,i-vin_start+pre_start+1,pre_end,vin,i+1,vin_end);
break;
}
}
return root;
}
};