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
本題使用一維線性規(guī)劃解決。
如果n等于0時(shí)，結(jié)果為0；
如果n等于1時(shí)，只有一個(gè)節(jié)點(diǎn)，結(jié)果為1；
如果n等于2時(shí)，根節(jié)點(diǎn)有兩種選擇，結(jié)果為2；
如果n大于3時(shí)，根節(jié)點(diǎn)有n種選擇，確定根節(jié)點(diǎn)后分別計(jì)算左右子樹(shù)的可能情況，然后相乘就是當(dāng)前根節(jié)點(diǎn)下所有的變形種類(lèi)，之后在求和即可。算法實(shí)現(xiàn)如下：
1 public class UniqueBinarySearchTrees {
2 public int numTrees(int n) {
3 if (n == 1)
4 return 1;
5 if (n == 2)
6 return 2;
7 int[] record = new int[n + 1];
8 record[0] = 1;
9 record[1] = 1;
10 record[2] = 2;
11 for (int i = 3; i <= n; i++) {
12 int tmp = 0;
13 for (int k = 0; k < i; k++) {
14 tmp += (record[k] * record[i - k - 1]);
15 }
16 record[i] = tmp;
17 }
18 return record[n];
19 }
20 }