给定节点数为 n 的二叉树的前序遍历和中序遍历结果,请重建出该二叉树并返回它的头结点。
例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建出如下图所示。
提示:
1.vin.length == pre.length
2.pre 和 vin 均无重复元素
3.vin出现的元素均出现在 pre里
4.只需要返回根结点,系统会自动输出整颗树做答案对比
数据范围:n \le 2000n≤2000,节点的值 -10000 \le val \le 10000−10000≤val≤10000
要求:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)
方法一:递归,每次递归都想不起怎么做,只是看得懂罢了
class Solution {
public:
TreeNode* reConstructBinaryTree(vector<int> pre,vector<int> vin) {
int size = vin.size();
if(size == 0){
return NULL;
}
TreeNode* root= new TreeNode(pre[0]);
int gen =0;
for(gen;gen<size;gen++){
if(vin[gen] == pre[0]){
break;
}
}
vector<int>pre_left,vin_left,pre_right,vin_right;
for(int i =0;i<gen;i++){
pre_left.push_back(pre[i+1]);
vin_left.push_back(vin[i]);
}
for(int i=gen+1;i<size;i++){
pre_right.push_back(pre[i]);
vin_right.push_back(vin[i]);
}
root->left = reConstructBinaryTree(pre_left,vin_left);
root->right = reConstructBinaryTree(pre_right,vin_right);
return root;
}
};
方法二: