题目链接:889. Construct Binary Tree from Preorder and Postorder Traversal
根据前序和后序建立一个二叉树,注意这种情况是不能唯一确定一棵树的,所以返回任意一个即可。
class Solution {
public:
TreeNode* constructFromPrePost(vector<int>& pre, vector<int>& post) {
return myBuildTree(pre,0,pre.size()-1,post,0,post.size()-1);
}
TreeNode* myBuildTree(vector<int>& pre, int preL, int preR, vector<int>& post, int poL, int poR) {
if(preL>preR||poL>poR)return NULL;
TreeNode* cur =new TreeNode(pre[preL]);
if (preL == preR) return cur; //必须加上,否则会无限递归
int i=0;
for(i=poL;i<=poR;i++){
if(post[i]==pre[preL+1]) break;//根据前根遍历的第二个数字,来将二叉树分成2段,它在后序遍历的位置左边的都是根节点的左子树,这里只是必要不充分条件,在它右边的也可能是根节点的左子树下的节点
}
cur->left = myBuildTree(pre,preL+1,preL+1+i-poL,post,poL,i);
cur->right= myBuildTree(pre,preL+1+i-poL+1,preR,post,i+1,poR-1);
return cur;
}
};