【题目】
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等于0时,结果为0;
如果n等于1时,只有一个节点,结果为1;
如果n等于2时,根节点有两种选择,结果为2;
如果n大于3时,根节点有n种选择,确定根节点后分别计算左右子树的可能情况,然后相乘就是当前根节点下所有的变形种类,之后在求和即可。
外循环是每个节点的情况都要算,保存在record[n]里面,record[i]里面记录的是当n=i的时候,BST可能性的个数
对于每一个record[i]的计算都是左子树的可能性*右子树的可能性
【代码】
public int numTrees(int n) {
if (n == 1)
return 1;
if (n == 2)
return 2;
int[] record = new int[n + 1];
record[0] = 1;
record[1] = 1;
record[2] = 2;
for (int i = 3; i <= n; i++) {
int tmp = 0;
for (int k = 0; k < i; k++) {
tmp += (record[k] * record[i - k - 1]);
}
record[i] = tmp;
}
return record[n];
}