34 leetcode - Combinations

最新推荐文章于 2025-08-13 21:50:19 发布

原创最新推荐文章于 2025-08-13 21:50:19 发布 · 242 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #Python #算法 #Combinate #组合

算法专栏收录该内容

84 篇文章

订阅专栏

本文介绍了一个经典的组合问题：给定两个整数 n 和 k，返回所有可能的 k 个数的组合，这些数来自 1 到 n 的整数集合。例如当 n=4 且 k=2 时，解决方案包括 [2,4]、[3,4] 等。通过递归方法实现了这一算法，并提供了一个完整的 Python 代码示例。

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

#!/usr/bin/python
# -*- coding: utf-8 -*-
'''
Combinations:Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.
For example,
If n = 4 and k = 2, a solution is:
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],  #左侧的数字要大于右侧
]
'''
class Solution(object):

    def combine(self, n, k):
        """
        :type n: int
        :type k: int
        :rtype: List[List[int]]
        """
        if k > n:
          return []
        ret = []
        self.__combine(n,k,1,0,range(k),ret)
        return ret

    def __combine(self,n,k,start_num,start_index,tmp,ret):
        if n - start_num + 1 < k - start_index:#小优化,如果剩余的数字的个数 < 要填充的个数,直接返回就好
            return
        if start_index == k:
            return ret.append(list(tmp))

        for i in range(start_num,n+1):
            tmp[start_index] = i
            self.__combine(n,k,i + 1,start_index + 1,tmp,ret)#右侧数字大于左侧数字,所以填充下一位时从左侧+1处开始即可

if __name__ == "__main__":
    s = Solution()
    print s.combine(4,2)