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
java代码,递归:
public static int numTrees(int n){
int num=0;
if (n==0)
return 0;
if(n==1)
return 1;
for(int i=0;i<=n-1;i++){
if(numTrees(i)!=0 && numTrees(n-i-1)!=0)
num += numTrees(i)*numTrees(n-i-1);
else
num += numTrees(i)+numTrees(n-i-1);
}
return num;
}
得学学eclipse调程序了!!!
去discuss逛了逛,说这是dp问题,有人给了代码:
/**
* Taking 1~n as root respectively:
* 1 as root: # of trees = F(0) * F(n-1) // F(0) == 1
* 2 as root: # of trees = F(1) * F(n-2)
* 3 as root: # of trees = F(2) * F(n-3)
* ...
* n-1 as root: # of trees = F(n-2) * F(1)
* n as root: # of trees = F(n-1) * F(0)
*
* So, the formulation is:
* F(n) = F(0) * F(n-1) + F(1) * F(n-2) + F(2) * F(n-3) + ... + F(n-2) * F(1) + F(n-1) * F(0)
*/
int numTrees(int n) {
int dp[n+1];
dp[0] = dp[1] = 1;
for (int i=2; i<=n; i++) {
dp[i] = 0;
for (int j=1; j<=i; j++) {
dp[i] += dp[j-1] * dp[i-j];
}
}
return dp[n];
}
晚上好好研究下,传说中的DP ~