题目地址:
https://leetcode.com/problems/unique-binary-search-trees/
问数值为 1 ∼ n 1\sim n 1∼n恰好 n n n个节点的二叉搜索树有多少个。
思路是动态规划。设有 f [ n ] f[n] f[n]个,则其左右子树节点数的可能性为 ( 0 , n − 1 ) , ( 1 , n − 2 ) , ( 2 , n − 3 ) , . . . , ( 0 , n − 1 ) (0,n-1),(1,n-2),(2,n-3),...,(0,n-1) (0,n−1),(1,n−2),(2,n−3),...,(0,n−1),所以 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 {
public int numTrees(int n) {
int[] dp = new int[n + 1];
dp[0] = 1;
for (int i = 1; i <= n; i++) {
for (int j = 0; j < i; j++) {
dp[i] += dp[j] * dp[i - j - 1];
}
}
return dp[n];
}
}
时间复杂度 O ( n 2 ) O(n^2) O(n2),空间 O ( n ) O(n) O(n)。
注解:
这其实就是Catalan数的递推公式。