题目内容
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=1时,ans=1
当n=2,ans=2
当n=3,ans=5
当n=4,ans=14
根据构建二叉查找树的特点,我们发现,当我们选择某个数为root结点的时候。则该root可以组成的结构情况为,所有left的情况×所有right的情况。所以我们可以构建dp的过程。
以n=5为例。当以1为root,假设dp[0]=1,则结果为count+=dp[0]*dp[4];当以2为root时,count+=dp[1]*dp[3];当以3为root时,count+=dp[2]*dp[2];当以4为root时,count+=dp[3]*dp[1];当以5为root时,count+=dp[4]*dp[0];
通过如上的过程我们可以知道状态转移方程为:dp[level]+=dp[level-root]*dp[root-1];
代码如下:
public class Solution {
public int numTrees(int n) {
int[] dp=new int[n+1];
dp[0]=1;
dp[1]=1;
for(int level=2;level<=n;level++)
for(int root=1;root<=level;root++){
dp[level]+=dp[level-root]*dp[root-1];
}
return dp[n];
}
}