96. 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
Subscribe to see which companies asked this question

注意：二分查找树的定义是，左子树节点均小于root，右子树节点均大于root！不要想当然地将某个点作为root时，认为其他所有节点都能全部放在left/right中，除非这个点是 min 或者 max 的。

分析：本题其实关键是递推过程的分析，n个点中每个点都可以作为root，当 i 作为root时，小于 i 的点都只能放在其左子树中，大于 i 的点只能放在右子树中，此时只需求出左、右子树各有多少种，二者相乘即为以 i 作为root时BST的总数。

如果n等于0时，结果为0；
如果n等于1时，只有一个节点，结果为1；
如果n等于2时，根节点有两种选择，结果为2；
如果n大于3时，根节点有n种选择，确定根节点后分别计算左右子树的可能情况，然后相乘就是当前根节点下所有的变形种类，之后在求和即可。

如果集合为空，只有一种BST，即空树，

UniqueTrees[0] =1

如果集合仅有一个元素，只有一种BST，即为单个节点UniqueTrees[1] = 1

UniqueTrees[2] = UniqueTrees[0] * UniqueTrees[1] (1为根的情况)+ UniqueTrees[1] * UniqueTrees[0] (2为根的情况。

再看一遍三个元素的数组，可以发现BST的取值方式如下：

UniqueTrees[3] = UniqueTrees[0]*UniqueTrees[2] (1为根的情况)

• UniqueTrees[1]*UniqueTrees[1] (2为根的情况)

• UniqueTrees[2]*UniqueTrees[0] (3为根的情况)

所以，由此观察，可以得出UniqueTrees的递推公式为UniqueTrees[i] = ∑ UniqueTrees[0…k] * [i-1-k] k取值范围 0<= k <=(i-1)

public class Solution {
    public int numTrees(int n) {

        if(n == 0)
            return 0;
        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];

    }
}