描述

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

示例:

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

Python

看题型知道应该使用backtracking遍历

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        res = []
        self.dfs(range(1,n+1),k,0,[],res)
        return res
    def dfs(self,nums,k,index,path,res):
        if k==0:
            res.append(path)
            return
        for i in range(index,len(nums)):
            self.dfs(nums,k-1,i+1,path+[nums[i]],res)

直接上python库

from itertools import combinations

class Solution:
    def combine(self, n, k):
        return list(combinations(range(1, n+1), k))