一、题目
给定两个整数 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
二、解法
回溯,先固定一个位置,然后遍历后续的所有位置,然后再回退
完整代码
class Solution:
def combine(self, n: int, k: int) -> List[List[int]]:
res = []
path = []
def backtrack(i):
if len(path) == k:
res.append(path.copy())
return
for j in range(i, n + 1):
path.append(j)
backtrack(j + 1)
path.pop()
backtrack(1)
return res