leetcode之unique-binary-search-trees(求长度为n的BST有多少个)
题目
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.
题意
给定一个常数n,从1-n开始的常数组成了一群BST树,求问有多少个这样不同的BST树。
解题思路
这里明显可以找到规律:
假设dp[i] 是常数i有多少个BST树,则:
dp[0] 空树 一个BST树
dp[1] 一个父节点 也是一个BST树
dp[2] 两个节点 = dp[0]*dp[1]+dp[1]dp[0] 也就是左子树为空+右子树为空,则为2个;
dp[3] 3个节点 = dp[0dp[2]+dp[1]*dp[1]+dp[2]*dp[0],即左子树为空、左右各一个、右子树为空这三种情况:
针对长度为i的数组:
dp[i] += dp[j]*dp[i-j-1] 这里,0=<j<i。 这就是常规递推公式:
C++实现代码
public class Solution {
public int numTrees(int n) {
if(n<0)
return -1;
int[] dp = new int[n+1];
dp[0]=1;
dp[1]=1;
for(int i=2;i<n+1;i++){
for(int j=0;j<i;j++){
dp[i] += dp[j]*dp[i-j-1];
}
}
return dp[n];
}
}