【Lintcode】164. Unique Binary Search Trees II

题目地址：

https://www.lintcode.com/problem/unique-binary-search-trees-ii/description

给定一个正整数$n$，返回以$1\sim n$为节点的所有二叉搜索树的树根。

开一个函数，返回所有以$l\sim r$为节点的BST的树根。那么可以从$l\sim r$来枚举树根的数值，如果树根数值确定，左右子树的树根的列表只需要递归调用就能得到。只需要注意当$l>r$的时候，也需要返回一个含null的列表，因为空树也算一棵BST。代码如下：

import java.util.ArrayList;
import java.util.List;

public class Solution {
    /**
     * @paramn n: An integer
     * @return: A list of root
     */
    public List<TreeNode> generateTrees(int n) {
        // write your code here
        return build(1, n);
    }
    
    private List<TreeNode> build(int l, int r) {
        List<TreeNode> res = new ArrayList<>();
        if (l > r) {
            res.add(null);
            return res;
        }
        
        // 枚举树根的取值
        for (int i = l; i <= r; i++) {
            List<TreeNode> left = build(l, i - 1), right = build(i + 1, r);
            for (TreeNode lRoot : left) {
                for (TreeNode rRoot : right) {
                    TreeNode root = new TreeNode(i);
                    root.left = lRoot;
                    root.right = rRoot;
                    res.add(root);
                }
            }
        }
        
        return res;
    }
}

class TreeNode {
    int val;
    TreeNode left, right;
    
    public TreeNode(int val) {
        this.val = val;
    }
}

时间复杂度可以由递推式$T(n)={\sum }_{i=0}^{n-1}T(i)T(n-i-1)$得到，其实就是Catalan数，所以$T(n)=O(\frac{4^n}{n^{\frac{3}{2}}})$。空间复杂度是$nT(n)=O(\frac{4^n}{n^{\frac{1}{2}}})$。