题目描述:
返回与给定的前序和后序遍历匹配的任何二叉树。
pre 和 post 遍历中的值是不同的正整数。
示例:
输入:pre = [1,2,4,5,3,6,7], post = [4,5,2,6,7,3,1]
输出:[1,2,3,4,5,6,7]
提示:
1 <= pre.length == post.length <= 30
pre[] 和 post[] 都是 1, 2, …, pre.length 的排列
每个输入保证至少有一个答案。如果有多个答案,可以返回其中一个。
方法1:
主要思路:
(1)使用前序遍历中的首个元素的后面一个元素,将后续遍历中的对应范围进行分割;
(2)递归的在分割后的范围内,重新建立新的树;
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* helper(vector<int>&pre,int pre_s,int pre_e,vector<int>&post,int post_s,int post_e){
if(pre_s>pre_e){//范围内无值
return NULL;
}
TreeNode* root=new TreeNode(pre[pre_s]);//当前结点作为根节点建立树
if(pre_s==pre_e){//若当前范围内只有一个值,则直接返回
return root;
}
//找出新的左右子树对应的范围
int index=post_s;
while(pre[pre_s+1]!=post[index]){
++index;
}
//建立左右子树
root->left=helper(pre,pre_s+1,pre_s+1+index-post_s,post,post_s,index);
root->right=helper(pre,pre_s+1+index-post_s+1,pre_e,post,index+1,post_e-1);
return root;
}
TreeNode* constructFromPrePost(vector<int>& pre, vector<int>& post) {
return helper(pre,0,pre.size()-1,post,0,post.size()-1);
}
};