894. 所有可能的真二叉树

894. 所有可能的真二叉树

题意

给你一个整数 n ，请你找出所有可能含 n 个节点的 真二叉树 ，并以列表形式返回。答案中每棵树的每个节点都必须符合 Node.val == 0 。

答案的每个元素都是一棵真二叉树的根节点。你可以按 任意顺序 返回最终的真二叉树列表。

真二叉树 是一类二叉树，树中每个节点恰好有 0 或 2 个子节点。

示例

输入：n = 7
输出：[[0,0,0,null,null,0,0,null,null,0,0],[0,0,0,null,null,0,0,0,0],[0,0,0,0,0,0,0],[0,0,0,0,0,null,null,null,null,0,0],[0,0,0,0,0,null,null,0,0]]
示例 2：

输入：n = 3
输出：[[0,0,0]]

题解

class Solution {
public:
    map<int , vector<TreeNode*>> ma;
    vector<TreeNode*> allPossibleFBT(int n) {
        //计算特殊情况，其中0，1，2，>2
        if(n==0||n==2){
            return {};
        }else if(n==1){
            return {new TreeNode(0)};
        }
        if(ma.count(n)!=0)
        {
            return ma[n];
        }
        vector<TreeNode*> ret;
        //分类计算，首先计算左节点，然后计算右节点，最后合并，其中左节点的种类为从1开始统计
        for(int i = 0;i<n-1;i++)//之所以是n-1，因为除去根节点和右节点的1个，剩下n-2
        {
            vector<TreeNode*> left= allPossibleFBT(i);
            vector<TreeNode*> right = allPossibleFBT(n-i-1);//1为根节点
            for(TreeNode* l:left)//一共的可能性是左子树*右子树,但是出现重复计算，其中左子树会和右子树重复
            {
                for(TreeNode* r:right){
                    TreeNode* root = new TreeNode(0);
                    root->left = l;
                    root->right = r;
                    ret.push_back(root);
                }
            }
        }
        ma[n] = ret;
        return ret;
    }
};