-
Notifications
You must be signed in to change notification settings - Fork 20
/
UniqueBinarySearchTrees.java
46 lines (39 loc) · 1.04 KB
/
UniqueBinarySearchTrees.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
/**
* Given n, how many structurally unique BST's (binary search trees) that store values 1...n?
* <p>
* For example,
* Given n = 3, there are a total of 5 unique BST's.
* <p>
* 1 3 3 2 1
* \ / / / \ \
* 3 2 1 1 3 2
* / / \ \
* 2 1 2 3
* <p>
* Accepted.
*/
public class UniqueBinarySearchTrees {
// Recursive solution. Time limit exceeded.
/*public int numTrees(int n) {
if (n == 0 || n == 1) {
return 1;
}
int r = 0;
for (int i = 1; i <= n; i++) {
r = r + numTrees(i - 1) * numTrees(n - i);
}
return r;
}*/
// Dynamic programming.
public int numTrees(int n) {
int[] array = new int[n + 2];
array[0] = 1;
array[1] = 1;
for (int i = 2; i <= n; i++) {
for (int j = 0; j < i; j++) {
array[i] += array[j] * array[i - j - 1];
}
}
return array[n];
}
}