LeetCode-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

 

思路

使用递归方法进行解题，调用 generateTrees(int star,int end) 方法在范围内遍历可作为根节点的数，并将初始范围划分为左右两部分，再继续调用 generateTrees(int star,int end) 方法。

最后将返回的左右子树拼接在根节点上，最终得到所有的二叉搜索树。

代码如下：

public class LeetCode_95 {
    
     public class TreeNode {
          int val;
          TreeNode left;
          TreeNode right;
          TreeNode() {}
          TreeNode(int val) { this.val = val; }
          TreeNode(int val, TreeNode left, TreeNode right) {
              this.val = val;
              this.left = left;
              this.right = right;
          }
      }
          
    public List<TreeNode> generateTrees(int n) {
        
        List<TreeNode> re = new ArrayList<LeetCode_95.TreeNode>();
        
        if(n==0) return re;
        
        re = generateTrees(1,n);
        
        return re;
    }
    
    
    public List<TreeNode> generateTrees(int star,int end){
        
        List<TreeNode> re = new ArrayList<LeetCode_95.TreeNode>();
        
        if(star>end) re.add(null);
        
        for(int i=star;i<=end;i++){
            
            List<TreeNode> leftTrees = generateTrees(star, i-1); // 获取所有左子树
            
            List<TreeNode> rightTrees = generateTrees(i+1, end); // 获取所有右子树
            
            for (TreeNode left : leftTrees) {     // 遍历左右子树，将其拼接在跟节点上
                for (TreeNode right : rightTrees) {
                    TreeNode currTree = new TreeNode(i);
                    currTree.left = left;
                    currTree.right = right;
                    re.add(currTree);
                }
            }


        }
              
        return re;
    }
    
}