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
思路:可视化之后,发现总数=依次取i为root之后,左边的可能性×右边的可能性。而可能性,是由node的数量来决定的,所以想到保存之前算过的值,这里就会想到是DP。注意一点:就是dp的array size是n+1, 因为要算有n的node的情况。dp表示的物理意义是,当前如果有n个node,可能性有多少。
- //由1,2,3,...,n构建的二叉查找树,以i为根节点,左子树由[1,i-1]构成,其右子树由[i+1,n]构成。
- //定义f(i)为以[1,i]能产生的Unique Binary Search Tree的数目
- //若数组为空,则只有一种BST,即空树,f(0)=1;
- //若数组仅有一个元素1,则只有一种BST,单个节点,f(1)=1;
- //若数组有两个元素1,2,则有两种可能,f(2)=f(0)*f(1)+f(1)*f(0);
- //若数组有三个元素1,2,3,则有f(3)=f(0)*f(2)+f(1)*f(1)+f(2)*f(0)
- //由此可以得到递推公式:f(i)=f(0)*f(i-1)+...+f(k-1)*f(i-k)+...+f(i-1)*f(0)
- //利用一维动态规划来求解
class Solution {
public int numTrees(int n) {
if(n <= 0) {
return 0;
}
int[] f = new int[n + 1];
f[0] = 1;
f[1] = 1;
for(int i = 2; i <= n; i++) {
for(int j = 0; j < i; j++) {
f[i] += f[j] * f[i - 1 - j];
}
}
return f[n];
}
}