【Leetcode刷题Python】77. 组合

最新推荐文章于 2024-03-27 08:30:00 发布

Better Bench

最新推荐文章于 2024-03-27 08:30:00 发布

阅读量817

点赞数

CC 4.0 BY-SA版权

分类专栏： LeeCode刷题文章标签： leetcode 算法递归

本文链接：https://blog.csdn.net/weixin_43935696/article/details/126126291

LeeCode刷题专栏收录该内容

152 篇文章

订阅专栏

该博客介绍了如何使用Python来生成从1到n的整数中所有可能的k个数的组合。提供了两种方法：一是利用内置的`itertools.combinations`函数；二是通过递归实现。示例展示了对于不同输入n和k的组合结果。

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

1 题目

给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。

你可以按任何顺序返回答案。

示例 1：

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

示例 2：

输入：n = 1, k = 1
输出：[[1]]

2 解析

（1）方法一
采用Python内置的排列组合方法 from itertools import combinations

（2）方法二
递归： #在 [1, n] 中选 k 个数，那么按照字典序可以在 [1, n - k +1]中依次选一个数x ，然后在 [x + 1, n] 中选出 (k - 1) 个数，通过递归进行搜索即可。

3 Python实现

（1）方法一

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        from itertools import combinations
        return [list(i) for i in combinations(iterable=range(1,n+1),r=k)]

（2）方法二

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        # 在 [1, n] 中选 k 个数
        # 那么按照字典序可以在 [1, n - k +1]中选一个数x 
        # 然后在 [x + 1, n] 中选出 (k - 1) 个数，通过递归进行搜索即可。
        
        def merge(array,k):
            if k==1:
                return [[i] for i in array]
            else:
                ans = []
                for i in range(len(array)-k+1):
                    last = merge(array[i+1:],k-1)
                    for j in last:
                        ans.append([array[i]]+j)
                return ans
        return merge(list(range(1, n+1)), k)