【Leetcode】889. Construct Binary Tree from Preorder and Postorder Traversal

题目链接:889. Construct Binary Tree from Preorder and Postorder Traversal
根据前序和后序建立一个二叉树,注意这种情况是不能唯一确定一棵树的,所以返回任意一个即可。

class Solution {
public:
    TreeNode* constructFromPrePost(vector& pre, vector& post) {
        return myBuildTree(pre,0,pre.size()-1,post,0,post.size()-1);
    }
    TreeNode* myBuildTree(vector& pre, int preL, int preR, vector& 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;
    }
};

你可能感兴趣的:(LEETCODE)