leetcode-77-组合题解

最新推荐文章于 2025-03-11 02:00:00 发布

Spaceack

最新推荐文章于 2025-03-11 02:00:00 发布

阅读量177

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Python 算法题解文章标签： leetcode python 算法

本文链接：https://blog.csdn.net/a541972321/article/details/113902013

Python 同时被 3 个专栏收录

75 篇文章

订阅专栏

题解

51 篇文章

订阅专栏

算法

24 篇文章

订阅专栏

该博客介绍了如何利用Python的itertools库来解决LeetCode中的组合问题。提供了使用combinations、combinations_with_replacement、permutations和product四个函数的示例，分别对应不放回、有放回的组合和排列，以及笛卡尔积的计算。通过这些函数，可以简洁地生成指定长度的数列组合和排列。

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

题目描述

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

示例

示例 1:
```
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
```
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/robot-return-to-origin

解题思路

虽然是中等题，但是使用python内置函数，就简单了。

组合,没有重复的情况（不放回抽样组合）使用 itertools.combinations 方法，

    def combine(self, n: int, k: int) -> List[List[int]]:
        result = []
        for i in itertools.combinations(list(range(1, n+1)), k):
            result.append(list(i))
        return result
        # [[1, 2], [1, 3], [1, 4], [2, 3], [2, 4], [3, 4]] 6

其它方法：

组合,有重复的情况（放回抽样组合）使用 itertools.combinations_with_replacement 方法，

    def combine(self, n: int, k: int) -> List[List[int]]:
        result = []
        for i in itertools.combinations_with_replacement(list(range(1, n+1)), k):
            result.append(list(i))
        return result
        # [[1, 1], [1, 2], [1, 3], [1, 4], [2, 2], [2, 3], [2, 4], [3, 3], [3, 4], [4, 4]] 10

排列　（不放回抽样排列）使用 itertools.permutations 方法，

    def combine(self, n: int, k: int) -> List[List[int]]:
        result = []
        for i in itertools.permutations(list(range(1, n+1)), k):
            result.append(list(i))
        return result
        # [[1, 2], [1, 3], [1, 4], [2, 1], [2, 3], [2, 4], [3, 1], [3, 2], [3, 4], [4, 1], [4, 2], [4, 3]] 12

笛卡尔积（有放回抽样排列）使用 itertools.product 方法，

    def combine(self, n: int, k: int) -> List[List[int]]:
        result = []
        for i in itertools.product(list(range(1, n+1)), repeat = k):
            result.append(list(i))
        return result
        # [[1, 1], [1, 2], [1, 3], [1, 4], [2, 1], [2, 2], [2, 3], [2, 4], [3, 1], [3, 2], [3, 3], [3, 4], [4, 1], [4, 2], [4, 3], [4, 4]] 16

combinations和permutations返回的是对象地址, 需要将iterator 转换成list 即可

题解1:

执行用时：48 ms, 在所有 Python3 提交中击败了95.61%的用户

内存消耗：14.8 MB, 在所有 Python3 提交中击败了98.26%的用户

import itertools
from typing import List

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