Leetcode_不同的二叉搜索树II

题目

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

 

题解

和上一题的思路类似。采用递归的方法分治得到两侧子树的所有情况。主要注意返回条件和递归参数。

vector<TreeNode*> GenerateTrees(int start, int end) {
        if(start > end)
            return {nullptr};
        vector<TreeNode*> ans;
        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){
                    TreeNode *p = new TreeNode(i, l, r);
                    ans.push_back(p);
                }
            }
        }
        return ans;
    }

oj调用的部分：

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

迭代更快一点。一开始本来想沿用上一题的做法，然而改变右子树的数值实在是太麻烦了，遂放弃。