题目地址:
https://www.lintcode.com/problem/unique-binary-search-trees/description
给定 1 ∼ n 1\sim n 1∼n共 n n n个数字为关键字,问能构成多少个二叉搜索树。
思路是动态规划。设 f [ n ] f[n] f[n]是 n n n对应的方案数,那么其按左右子树有多少个节点可以分为 ( 0 ) + ( n − 1 ) , ( 1 ) + ( n − 2 ) , ( 2 ) + ( n − 3 ) , . . . , ( n − 1 ) + ( 0 ) (0)+(n-1),(1)+(n-2),(2)+(n-3),...,(n-1)+(0) (0)+(n−1),(1)+(n−2),(2)+(n−3),...,(n−1)+(0)共 n n n种情况,所以 f [ n ] = ∑ i = 0 n − 1 f [ i ] f [ n − 1 − i ] f[n]=\sum_{i=0}^{n-1}f[i]f[n-1-i] f[n]=i=0∑n−1f[i]f[n−1−i]而 f [ 0 ] = 1 f[0]=1 f[0]=1,然后递推过去即可。代码如下:
public class Solution {
/**
* @param n: An integer
* @return: An integer
*/
public int numTrees(int n) {
// write your code here
int[] dp = new int[n + 1];
dp[0] = 1;
for (int i = 1; i <= n; i++) {
for (int j = 0; j <= i - 1; j++) {
dp[i] += dp[j] * dp[i - 1 - j];
}
}
return dp[n];
}
}
时间复杂度 O ( n 2 ) O(n^2) O(n2),空间 O ( n ) O(n) O(n)。
注解:
这其实就是Catalan数。可以参考https://blog.csdn.net/qq_46105170/article/details/104243540。