题目描述
分析
由题可得,真二叉数的结点数量必定为奇数。
首先能想到用递归来构建每一种可能的情况。那么每次递归时应该怎样进行选择呢?
一种策略是:每次都从度为0的结点中选择一个,给它填充左右子结点。但是这样会导致最终答案重复。
另一种是策略是:每次都从当前结点选择其左右子树结点的数量,如果需要从当前结点开始构建一个n个结点的真二叉树,那么左右子树的结点总数为n-1,左右子树的结点数量也都为奇数,因此可以将左子树的结点数量从1开始每次加2进行遍历,这样就可以枚举出所有情况。当n==1时就可以结束递归。
#include <vector>
#include <deque>
using namespace std;
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<TreeNode*> allPossibleFBT(int n) {
vector<TreeNode *> res;
if (n == 1) return { new TreeNode };
for (int left = 1; left < n; left += 2)
{
vector<TreeNode *> leftTree = allPossibleFBT(left);
vector<TreeNode *> rightTree = allPossibleFBT(n - left - 1);
for (int i = 0; i < leftTree.size(); ++i)
{
for (int j = 0; j < rightTree.size(); ++j)
{
TreeNode *node = new TreeNode;
node->left = leftTree[i];
node->right =rightTree[j];
res.push_back(node);
}
}
}
return res;
}
};