一、题目描述
给定两个整数 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方法进行转换。