【树】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

解答:

/**
 * Definition for a binary tree node.
 * 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;
 *     }
 * }
 */
class Solution {
    public List<TreeNode> generateTrees(int n) {
        if(n<=0){
            return new LinkedList<TreeNode>();
        }

        return helper(1,n);
    }

    public List<TreeNode> helper(int start, int end){
        List<TreeNode> res = new LinkedList<>();
        
        if(start>end){
            res.add(null);
            return res;
        }

        for(int i = start;i<=end;i++){
            List<TreeNode> left = helper(start,i-1);
            List<TreeNode> right = helper(i+1,end);

            for(TreeNode l:left)
                for(TreeNode r:right){
                    TreeNode root = new TreeNode(i);
                    root.left = l;
                    root.right = r;

                    res.add(root);
                }

        }

        return res;
    }
}

 

posted @ 2020-09-22 21:26  3KBLACK  阅读(60)  评论(0)    收藏  举报