糖逗 发表于 2020-5-13 14:07:20

C++刷leetcode（95. 不同的二叉搜索树 II）【递归】

题目描述：

给定一个整数 n，生成所有由 1 ... n 为节点所组成的二叉搜索树。

示例:

输入: 3
输出:
[
,
,
,
,

]
解释:
以上的输出对应以下 5 种不同结构的二叉搜索树：

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


/**
* Definition for a binary tree node.
* struct TreeNode {
*   int val;
*   TreeNode *left;
*   TreeNode *right;
*   TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
    vector<TreeNode*> dfs(int start, int end){
      vector<TreeNode*> res;
      if(start > end){
            res.push_back(NULL);
            return res;
      }
      else if(start == end){
            TreeNode* root = new TreeNode(start);
            res.push_back(root);
            return res ;
      }
      else{
            for(int i = start; i <= end; i++){
                vector<TreeNode*> left = dfs(start, i-1);
                vector<TreeNode*> right = dfs(i+1, end);
                for(auto ch1 : left){
                  for(auto ch2: right){
                        TreeNode*root = new TreeNode(i);
                        root -> left = ch1;
                        root -> right = ch2;
                        res.push_back(root);
                  }
                }
            }
      }
      return res;
    }
    vector<TreeNode*> generateTrees(int n) {
      if(n == 0){
            vector<TreeNode*> res;
            return res;
      }
      else return dfs(1, n);
    }
};


参考链接：https://leetcode-cn.com/problems/unique-binary-search-trees-ii/solution/xiang-xi-tong-su-de-si-lu-fen-xi-duo-jie-fa-by-2-7/

糖逗 发表于 2020-5-13 14:07:58

需要再刷一下，没写出来{:10_266:}

永恒的蓝色梦想 发表于 2020-5-13 14:08:16

一直搞不懂二叉树……{:10_277:}

糖逗 发表于 2020-5-13 14:09:51

这道题结合https://fishc.com.cn/thread-168613-1-1.html 再看一下

糖逗 发表于 2020-5-13 14:12:41

永恒的蓝色梦想 发表于 2020-5-13 14:08
一直搞不懂二叉树……

可能多做几道二叉树的问题会好一些{:10_266:}
页: [1]
查看完整版本: C++刷leetcode（95. 不同的二叉搜索树 II）【递归】