95. 不同的二叉搜索树 II

原题链接：95. 不同的二叉搜索树 II

 

solution：

①在区间[l,r]枚举每个以i为根节点的二叉搜索树，用left保存区间[l,i - 1]构造左子树的集合, 用right保存区间[i + 1,r]构造右子树的集合.

② left中保存的每一个左子树的情况都可以和right中保存的右子树的情况构成一个新的二叉搜索树，假设left中有n种,right中有m种，总的方案就有n * m种

③递归求解

class Solution {
public:
    vector<TreeNode *> dfs(int l,int r) {
        vector<TreeNode *> res;
        if(l > r) {
            res.push_back(nullptr);
            return res;
        } 

        for(int i = l;i <= r;i++) {
            vector<TreeNode *> left = dfs(l,i - 1); //左子树序列
            vector<TreeNode *> right = dfs(i + 1,r); //右子树序列

            for(auto &lnode :left)
                for(auto *rnode : right) {
                    TreeNode *root = new TreeNode(i);
                    root->left = lnode;
                    root->right = rnode;
                    res.push_back(root);
                }
        }
        return res;
    }

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