原题链接 https://leetcode.com/problems/unique-binary-search-trees/
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
首先分析一下
1 1 2
\ /2 1
这是1和2的情况。
对于n的情况
n
/ \
小于n 大于n的
如果n == 1,则小于n的情况就是0的情况,大于n的情况就是n - 1的情况,这种情况的总数为 ans[0] * ans[n - 1];
如果n == 2, 则小于n的情况 就是1的情况,大于n的情况就是n - 2,这种情况的总数为ans[1] * ans[n-1];
...
如果n == n 则小于n的情况就是 n- 1的情况,大于n的情况就是0的情况,这种情况下的总数为ans[n - 1] * ans[0];
那么ans[0]* ans[n - 1] 和 ans[n-1]*ans[0]的情况是否相同叻,答案是否定的。
因此 ans[i] = sum(ans[j] * ans[i - j - 1]) , 1<=i<= n, 0<=j <= i -1. ans[0] = 0;
class Solution {
public:
int numTrees(int n) {
vector<long long> ans(n + 3, 0);
ans[0] = 1; ans[1] = 1; ans[2] = 2;
for (int i = 3; i < n + 1; ++i)
{
for (int j = 0; j <= i - 1; ++j)
ans[i] += ans[j] * ans[i - j - 1];
}
return ans[n];
}
};