题目描述:
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
题目分析:
这是一道DP的Medium难度的题目。
要构造出不同的二叉搜索树,首先要找一个数作为树根。然后根左子树中的每一个数一定都比根的值小,根右子树中的每一个数一定都比根的值大。可以由此得到DP的第一个推算式:sum[i] = f[1, i] +f[2, i] + …… + f[i, i],分别表示以1–i的每一个不同数作为树根的情况各有多少种。确定了树根以后,左子树可以递归的表示为sum[i - 1],右子树可以递归的表示为sum[n - i],因此得到第二个DP推算式:f[j, i] = sum[j - 1] + sum[i - j]。
具体步骤:
- step1:首先sum[0]和sum[1]值为1,因为没有根或根只有一个数都只有一种情况。
- step2:往后对于每一个i值,通过sum[i] = f[1, i] +f[2, i] + …… + f[i, i]求值。
- step3:而对于每一个f[j, i],通过f[j, i] = sum[j - 1] + sum[i - j]求值。
- step4:最后返回sum[n]的值。
代码:
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int numTrees(int n) {
vector<int> v(n + 1, 0); //建立数组记录i个数时,BST的种数
v[0] = v[1] = 1; //没有根或者根只有一个数时,都只有一种
for (int i = 2; i <= n; i++){ //遍历n = i,求出每个i对应的情况总数
for (int j = 1; j <= i; j++){ //v[i] = f[1, i] + f[2, i] + …+ f[i, i]
v[i] += v[j - 1] * v[i - j]; //f[j, i] = v[j - 1] + v[i - j]
}
}
return v[n]; //返回v[n]
}
};