LeetCode 77. 组合（C++、python）

最新推荐文章于 2022-08-02 17:35:00 发布

原创最新推荐文章于 2022-08-02 17:35:00 发布 · 300 阅读

1 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

713 篇文章

订阅专栏

回溯

25 篇文章

订阅专栏

本文深入解析了组合生成算法，通过示例展示了如何使用C++和Python实现从1到n的所有可能k个数的组合。提供了详细的代码实现，帮助读者理解递归深度优先搜索(DFS)在组合生成问题中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

C++

class Solution {
public:
    void DFS(vector<vector<int>>& res, vector<int>& tmp, int n,int k, int w)
    {
        if(k==tmp.size())
        {
            res.push_back(tmp);
            return;
        }
        for(int i=w;i<=n;i++)
        {
            if(tmp.size()<k)
            {
                tmp.push_back(i);
                DFS(res,tmp,n,k,i+1);
                tmp.pop_back();                
            }
        }
    }
    
    vector<vector<int>> combine(int n, int k) 
    {
        vector<vector<int>> res;
        vector<int> tmp;
        DFS(res,tmp,n,k,1);
        return res;
    }
};

python

class Solution:
    def DFS(self, res, tmp, n, k, w):
        if k==len(tmp):
            res.append(tmp.copy())
            return
        for i in range(w,n+1):
            if len(tmp)<k:
                tmp.append(i)
                self.DFS(res,tmp,n,k,i+1)
                del tmp[-1]
    
    def combine(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: List[List[int]]
        """
        res=[]
        tmp=[]
        self.DFS(res,tmp,n,k,1)
        return res