题目地址:
https://leetcode.com/problems/all-possible-full-binary-trees/
返回所有 N N N个节点的full binary tree的树根。所有节点都取值为 0 0 0。
思路是递归。首先如果 N > 0 N>0 N>0是偶数,那么不存在(可以用数学归纳法证明所有的full binary tree都是奇数个节点)。节点个数是 1 1 1的树也是full binary tree,很好构造。接着取所有小于 N N N的奇数 k k k,先构造所有节点个数 k k k的full binary tree,然后构造所有节点个数 N − 1 − k N-1-k N−1−k的full binary tree,然后new出树根,两层for循环,将左子树右子树连到树根上。代码如下:
import java.util.ArrayList;
import java.util.List;
public class Solution {
public List<TreeNode> allPossibleFBT(int N) {
List<TreeNode> res = new ArrayList<>();
if (N % 2 == 0) {
return res;
}
return dfs(N);
}
private List<TreeNode> dfs(int N) {
List<TreeNode> res = new ArrayList<>();
if (N == 1) {
res.add(new TreeNode(0));
return res;
}
for (int i = 1; i < N - 1; i += 2) {
List<TreeNode> left = dfs(i), right = dfs(N - 1 - i);
for (TreeNode l : left) {
for (TreeNode r : right) {
TreeNode root = new TreeNode(0);
root.left = l;
root.right = r;
res.add(root);
}
}
}
return res;
}
}
class TreeNode {
int val;
TreeNode left, right;
public TreeNode(int val) {
this.val = val;
}
}
时空复杂度 O ( C n ) O(C_n) O(Cn),其中 C n C_n Cn是Catalan数。只需注意到 T ( n ) = T ( 1 ) T ( n − 2 ) + T ( 3 ) T ( n − 4 ) + . . . + T ( n − 2 ) T ( 1 ) T(n)=T(1)T(n-2)+T(3)T(n-4)+...+T(n-2)T(1) T(n)=T(1)T(n−2)+T(3)T(n−4)+...+T(n−2)T(1)接着按照类似求Catalan数的方法来求一下 T ( n ) T(n) T(n)的表达式即可。