Given n, how many structurally unique BST's (binary search trees) that store values 1...n?
For example,
Given n = 3, there are a total of 5 unique BST's.
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
寻找总结点数为n的构建二叉树的可能数目。这种题肯定是递归或者动态规划,明显的子问题拆解。比如例子中的5, 你把1放在root,剩下的只能放在右侧,变为规模为4的问题,如果把3放在中间,俩在左,三个在右面,变成了规模2的结果乘上规模为3的结果。
所以可以构建dp[n+1],其中dp[0]=1, dp[1]=1。递归公式为:
这个公式也就是卡塔兰数的递推表达,具体见链接里的介绍。总之这种递推解决方法还是比较好想到的,具体代码如下:
时间复杂发O(n^2),空间复杂度O(n).
int numTrees(int n) {
int dp[n+1] = {0};
if (n == 0) return 0;
dp[0] = 1;
dp[1] = 1;
for (int i = 2; i < n+1; i++) {
for (int j = 0; j < i; j++) {
dp[i] += dp[j] * dp[i-j-1];
}
}
return dp[n];
}