77. 组合 Python

---

一、题目描述

给定两个整数 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]]

提示：
1 <= n <= 20
1 <= k <= n

二、代码

代码如下：

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

---

三、解题思路

本题本质上是一个数组的组合问题，有很多快速的组合方法，但是考虑到本题所需组合的数组nums的长度并不是很大（<=20），所以采用了itertools库中的combinations全组合方法快速得出组合数组列表。
需要注意有：①在生成待组合列表nums时所用的range方法，在python3中range返回的是类对象，需要用list()函数转成list；②combinations方法得到的答案并不是list列表，要进行转化，另外，转化后的数组list中的元素并不是[X,Y]类型,而是(X,Y)类型,也需要用list方法进行转换。