不同的二叉搜索树 II

用分治递归生成所有合法的二叉搜索树: 在区间[left, right],我们依次选每个值i作为根节点,递归生成左子树区间和右子树区间的所有可能结构,再将左、右子树的所有组合与根节点拼接,得到当前区间的所有二叉搜索树。

class Solution {
public:
    vector<TreeNode*> generateTrees(int n) {
        return dfs(1, n); // 从区间[1,n]开始递归
    }

    // 生成区间[left, right]对应的所有二叉搜索树
    vector<TreeNode*> dfs(int left, int right) {
        if (left > right) return {nullptr}; // 空树
        vector<TreeNode*> trees;

        // 选i作为当前根节点
        for (int i = left; i <= right; ++i) {
            // 递归生成左、右子树的所有可能
            vector<TreeNode*> l = dfs(left, i-1);
            vector<TreeNode*> r = dfs(i+1, right);
            
            // 与根i拼接
            for (TreeNode* lefttree : l) {
                for (TreeNode* righttree : r) {
                    TreeNode* root = new TreeNode(i);
                    root->left = lefttree;
                    root->right = righttree;
                    trees.push_back(root);//存入数组
                }
            }
        }
        return trees;
    }
};
全部评论

相关推荐

不愿透露姓名的神秘牛友
12-16 15:57
小鹏汽车 java后端 22*15(固定13,2个月年终) 硕士211
点赞 评论 收藏
分享
11-16 01:13
宜春学院 Java
点赞 评论 收藏
分享
11-04 19:05
已编辑
东莞城市学院 单片机
不知道怎么取名字_:你这个要实习两年?哪有这么久的,感觉就是即使你毕业了,但还按实习的话,是不是不用给你缴社保公积金啥的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务