LeetCode-95-不同的二叉搜索树II

LeetCode-95-不同的二叉搜索树II

题目

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

 给定一个整数 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

思路解析

由于之前做过这道题的上一题，96.不同的二叉搜索树，当时采用的非递归，所以我就顺其自然的考虑非递归解决问题。这里有一个特别麻烦的地方，在第96题的解释中，我描述到，当添加右子树的时候，会出现差值，当时只需要获取数量，所以可以不用管，但是这次需要输出搜索树，就不得不管了。

我的解决方案是，将右子树加上根节点的值，但是实际操作的时候遇到了浅层拷贝的问题，就是我需要复制出右子树的结构体，但是直接用=赋值的话，结构体中的指针变量所指向的值并不会拷贝，导致后续会出错；

但是要自己写一个深度拷贝感觉有点麻烦，背离了刷leetcode的本意，于是我瞄了一眼答案，

看到了大佬写的递归的代码，感叹我应该超越不了了，干脆偷懒到底直接提交。。。。

代码

class Solution {
public:
    vector<TreeNode*> generateTrees(int n) {
        if(n == 0) return {};
        return helper(1, n);
    }
    vector<TreeNode*> helper(int start, int end) {
        if(start > end) return {nullptr};
        vector<TreeNode*> res;
        for(int i=start; i<=end; i++) {
            auto left = helper(start, i-1), right = helper(i+1, end);
            for(auto a:left) {
                for(auto b:right) {
                    TreeNode* node = new TreeNode(i);
                    node->left = a;
                    node->right = b;
                    res.push_back(node);
                }
            }
        }
        return res;
    }
};