题目地址:
https://www.lintcode.com/problem/all-possible-full-binary-trees/description
返回所有节点数为 n n n的Full Binary Tree,即每个节点要么没有孩子,要么有 2 2 2个孩子。返回树根即可。所有节点的数值都赋值为 0 0 0。
思路是递归。首先,如果 n n n是偶数,那么是不存在这样的树的(空树除外),因为节点数等于总度数减去 1 1 1,而总度数是偶数。接着,这样的树的左右子树满足相同的性质,所以可以用递归。只需要枚举左右子树的节点数即可,从 1 1 1开始枚举并且只枚举奇数,然后接到新new出来的树根两边即可。代码如下:
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
public class Solution {
/**
* @param n: An integer
* @return: A list of root
*/
public List<TreeNode> allPossibleFBT(int n) {
// write your code here
if (n % 2 == 0) {
return new ArrayList<>();
}
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 - 2; i += 2) {
// 递归得到左右子树,然后用两重循环接到new出来的节点两边
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)的表达式即可。