leetcode 894. 所有可能的满二叉树

目录

• 题目描述：
• 示例：
• 解法：

题目描述：

满二叉树是一类二叉树，其中每个结点恰好有 0 或 2 个子结点。

返回包含 N 个结点的所有可能满二叉树的列表。 答案的每个元素都是一个可能树的根结点。

答案中每个树的每个结点都必须有 node.val=0。

你可以按任何顺序返回树的最终列表。

示例：

输入：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]]
解释：

提示：

• 1 <= N <= 20

---

解法：

/**
 * 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:
    vector<TreeNode*> allPossibleFBT(int N) {
        vector<vector<TreeNode*>> lst(N+1);
        lst[1] = {new TreeNode(0)};
        for(int i = 3; i <= N; i += 2){
            for(int j = 1; j < N; j += 2){
                if(i - 1 - j > 0){
                    for(TreeNode* left : lst[j]){
                        for(TreeNode* right : lst[i - j - 1]){
                            TreeNode* root = new TreeNode(0);
                            root->left = left;
                            root->right = right;
                            lst[i].push_back(root);
                        }
                    }
                }
            }
        }
        return lst.back();
    }
};