【leetcode】leetcode95不同的二叉搜索树 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.
 * 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 n) {
        vector<TreeNode*>ans;
        if(n==0)
        return ans;
        ans=CreateTree(1,n);
        return ans;
    }
    vector<TreeNode*> CreateTree(int begin,int end)
    {
        vector<TreeNode*>CurrentTree;
        if(begin>end)
        {
            CurrentTree.push_back(NULL);
            return CurrentTree;
        }
        vector<TreeNode*>LeftTree;
        vector<TreeNode*>RightTree;
        for(int i=begin;i<=end;i++)
        {
            //TreeNode* node=new TreeNode(i);//刚开始写在这里有问题
            LeftTree=CreateTree(begin,i-1);
            RightTree=CreateTree(i+1,end);
            for(int j=0;j<LeftTree.size();j++)
            for(int k=0;k<RightTree.size();k++)
            {
                TreeNode* node=new TreeNode(i);
                node->left=LeftTree[j];
                node->right=RightTree[k];
                CurrentTree.push_back(node);
            }
        }
        return CurrentTree;
    }
};

解题思路

二叉树的问题一般都是用递归来解，而这道题也是用递归来求解，感觉像是枚举所有的满足要求的二叉树？？但是需要注意的一点是，这里不是枚举所有的情况，而是枚举所有的二叉搜索树的情况，所以递归构建子树。每次构建一个子树的时候，都是首先创建当前结点，再构建当前结点的左子树（注意左子树上的结点一定要都小于当前结点的值）以及右子树（右子树上的结点一定要都大于当前结点的值）。因为每层递归都会产生很多个子树，为了存储所有的子树需要返回return一个vector而不是TreeNode*.
遇到的坑：刚开始我把TreeNode* node=new TreeNode(i);写在内层两个循环的上面，发现输出结果有问题，后来发现问题是因为，写在上面的话，就只创建了一个结点，所以最后得到的子树都是一样的（因为只new了一个结点，所以之后所有的更改都在这个上面。）正确的写法是应该为每种情况创建一个当前结点node，这样才能保证子树互不干扰，最后返回多个子树。