【力扣】[动态规划] 95.不同的二叉搜索树 II

1、题目

给你一个整数 n ，请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。可以按 任意顺序 返回答案。

链接： https://leetcode-cn.com/problems/unique-binary-search-trees-ii/

 

2、思路分析

1）回溯
我们定义 generateTrees(start, end) 函数表示当前值的集合为 [start,end]，返回序列 [start,end] 生成的所有可行的二叉搜索树。

我们考虑枚举 [start,end] 中的值 i 为当前二叉搜索树的根，那么序列划分为[start,i−1] 和[i+1,end] 两部分。我们递归调用这两部分，即 generateTrees(start, i - 1) 和 generateTrees(i + 1, end)，获得所有可行的左子树和可行的右子树。

那么最后一步我们只要从可行左子树集合中选一棵，再从可行右子树集合中选一棵拼接到根节点上，并将生成的二叉搜索树放入答案数组即可。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<TreeNode*> generateTrees(int start, int end)
    {
        if(start > end){
            return {nullptr};
        }
        if(start == end){
            return {new TreeNode(start)};
        }
        vector<TreeNode*> res;
        for(int i = start; i <= end; ++i){
            // 左子树集合
            vector<TreeNode*> left = generateTrees(start, i - 1);
            vector<TreeNode*> right = generateTrees(i + 1, end);
            for(auto & l :left){
                for(auto& r : right){
                    res.push_back(new TreeNode(i, l, r));
                }
            }
        }
        return res;
    }

    vector<TreeNode*> generateTrees(int n) {
        return generateTrees(1, n);
    }
};