[LeetCode] 95. Unique Binary Search Trees II 独一无二的二叉搜索树之二

题目

 

Given an integer n, generate all structurally unique BST's (binary search trees) that store values 1 ... n.

Example:

Input: 3
Output:
[
  [1,null,3,2],
  [3,2,null,1],
  [3,1,null,null,2],
  [2,1,3],
  [1,null,2,null,3]
]
Explanation:
The above output corresponds to the 5 unique BST's shown below:

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

 问题分析

本题要求求出从1~n的所有的二叉搜索树，以每个数作为根节点，使用分治的方式左右划分子树，递归构造 刚开始将[1,n]看作一个整体,需要将其每个数作为根节点,划分整个节点的左右区间,调用递归函数获取这个数的所有的左右子树,使用二重循环将所有的左右子树拼接到这个数构成的节点上,然后向上层返回,代码如下

 

代码示例

public List<TreeNode> generateTrees(int n) {

        if(n < 1){
           return new ArrayList<TreeNode>();
        }

        return help(1, n);
    }

    /**
     * 分治递归求解所有树
     * @param start
     * @param end
     * @return
     */
    private List<TreeNode> help(int start, int end) {

        List<TreeNode> res = new ArrayList<TreeNode>();
        // return null node
        if(start > end){
            res.add(null);
            return res;
        }
        // 从start到end每个节点都作为root向上返回，分治思想
        for(int i = start; i <= end; i++){

            // 左右子树
            List<TreeNode> left = help(start, i - 1);
            List<TreeNode> right = help(i + 1, end);
            // 将以该节点为root的所有左右子树拼接到该节点上，使用二重循环
            for(TreeNode a : left){

                for(TreeNode b : right){

                    TreeNode node = new TreeNode(i);
                    node.left = a;
                    node.right = b;
                    res.add(node);
                }
            }
        }

        return res;
    }