刷题---树篇---95. 不同的二叉搜索树Ⅱ

95. 不同的二叉搜索树 II

给定一个整数 n，生成所有由 1 ... n 为节点所组成的 二叉搜索树 。

 

示例：

输入：3
输出：
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
解释：
以上的输出对应以下 5 种不同结构的二叉搜索树：

   1         3     3      2      1
    \       /     /      / \      \
     3     2     1      1   3      2
    /     /       \                 \
   2     1         2                 3
 

提示：

0 <= n <= 8

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/unique-binary-search-trees-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

————————————————————————————————————————————————————————

类似题：

刷题---树篇---96. 不同的二叉搜索树

与96题类似的题目，我们可能会想到用dp来做。但是与96不同，本题需要给出具体的bst形式，所以实际上用dp是更复杂的。

既然是生成bst，那就应该从bst的性质入手分析。

按照96的思路，我们可以在1->n的节点中选出一个i作为根节点，然后将1->i-1 划分为右子树部分，i+1->n 划分为左子树部分，然后递归进行生成。

所以就能产生一个大体的算法框架:

func bst(i,n) {

        for k = i to n : 

             选择k作为根节点

             k.left = bst(i,k-1)   //右子树递归

             k.right = bst(k+1,n) //左子树递归

}

现在确定返回值：

我们需要将所有的中间生成的bst都保存下来，所以bst（）函数需要返回一个数组，保存所有的值；

确定边界值：

i < k时，返回一个值为nil的数组，i == n 时，此时是一个叶子节点，返回一个仅包含此节点的数组。

代码如下：

func generateTrees(n int) []*TreeNode {
    res := make([]*TreeNode,0)
    if n == 0 {
        return res
    }
    
    res = getBST(1,n)


    return res
}

func getBST(i,n int) []*TreeNode {
    a := make([]*TreeNode,0)
    if i > n {
        //放入一个nil值，可以进入for...range循环，防止遗漏
        t := &TreeNode{0,nil,nil}
        t = nil 
        a = append(a,t)
        return a
    } 
    if i == n {
        t := &TreeNode{i,nil,nil}
        a = append(a,t)
        return a
    }
    
    for root := i; root <= n; root++ {
        //左边
		l := getBST(i, root-1)
        //右边
		r := getBST(root+1, n)  
        //左右排列组合，和96题中的左右相乘本质一样
		for _, treeleft := range l {
			for _, treeright := range r {
				t := &TreeNode{root,nil,nil}
				t.Left = treeleft
				t.Right = treeright
				a = append(a,t)
			}
		}
	}
	return a
}