Leetcode 77. 组合 DFS+回溯+剪枝

原创已于 2025-09-08 17:40:19 修改 · 277 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #剪枝 #leetcode #算法 #数据结构

于 2022-05-07 01:25:21 首次发布

Leetcode 专栏收录该内容

377 篇文章

订阅专栏

本文详细介绍了LeetCode中的第77题——组合问题，提供了一种使用深度优先搜索（DFS）的C++解决方案。代码中展示了如何通过递归实现从1到n中选择k个数的所有组合，并进行了剪枝优化以避免无效的遍历。同时，该问题探讨了回溯法在解决组合问题中的应用。

原题链接：Leetcode 77. 组合

在这里插入图片描述
回溯1：枚举下一个数选哪个

class Solution {
public:
    vector<vector<int>> res;
    void dfs(int l, int r, int k, vector<int>& path) {
        if (k == 0) {
            res.push_back(path);
            return;
        }
        if (r - l + 1 < k)
            return; // 剪枝
        for (int i = l; i <= r; i++) {
            path.push_back(i);
            dfs(i + 1, r, k - 1, path);
            path.pop_back();
        }
    }
    vector<vector<int>> combine(int n, int k) {
        vector<int> path;
        dfs(1, n, k, path);
        return res;
    }
};

回溯2：选 / 不选

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;
    void dfs(int n,int k,int index){
        if(k==0){
            res.push_back(path);
            return ;
        }
        if(n-index+1<k) return ;
        if(index<=n){
            path.push_back(index);
            dfs(n,k-1,index+1);
            path.pop_back();
            dfs(n,k,index+1);
        }
    }
    vector<vector<int>> combine(int n, int k) {
        dfs(n,k,1);
        return res;
    }
};