LeetCode 95. Unique Binary Search Trees II

最新推荐文章于 2019-04-19 22:22:37 发布

15wylu

最新推荐文章于 2019-04-19 22:22:37 发布

阅读量227

点赞数

本文链接：https://blog.csdn.net/qq_32767041/article/details/86479475

版权

Leetcode 专栏收录该内容

53 篇文章 0 订阅

订阅专栏

本文介绍了一种使用递归方法生成所有结构上唯一的二叉搜索树的算法。通过遍历所有可能作为根节点的情况，递归构建左子树和右子树，实现了LeetCode题目Unique Binary Search Trees II的解决方案。

摘要由CSDN通过智能技术生成

Description

https://leetcode.com/problems/unique-binary-search-trees-ii/
给定n，生成所有结构上唯一的二叉搜索树，并返回所有这些树的root。

Solving Ideas

与Unique Binary Search Trees类似，不同的是，这里需要构造出所有结构上唯一的BST

实现起来比较简单就是使用递归的方法，遍历所有可能为root的情况，然后递归构建其左子树和右子树。

Solution

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<TreeNode> generateTrees(int n) {
        if (n == 0) {
            return new LinkedList<>();
        }
        return generate(1, n);
    }

    private LinkedList<TreeNode> generate(int start, int end) {
        LinkedList<TreeNode> allTrees = new LinkedList<>();
        if (start > end) {
            allTrees.add(null);
            return allTrees;
        }
        
        //pick up a root
        for (int i = start; i <= end; i++) {
            //all possible left subtrees if i is chosen to be a root
            LinkedList<TreeNode> leftTrees = generate(start, i - 1);
            
            //all possible right subtrees if i is chosen to be a root
            LinkedList<TreeNode> rightTrees = generate(i + 1, end);
            
            //connect left and right trees to the root i
            for (TreeNode left : leftTrees) {
                for (TreeNode right : rightTrees) {
                    TreeNode curTreeRoot = new TreeNode(i);
                    curTreeRoot.left = left;
                    curTreeRoot.right = right;
                    //put current tree root to the list
                    allTrees.add(curTreeRoot);
                }
            }
        }
        return allTrees;
    }
}