题目:
给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combinations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
类似之前的全排列.
回溯:
每次把n-1记录到数组.记录k-1,保证是k个数的组合.一旦k为0,就将记录n-1的数组加入答案.
此时回溯.
注意排列组合期间,曾经用过的数就不能再用了.我用的是递减回溯,因此只要新加入的元素大于记录数组的最后一个元素,就要跳过这次排列.
代码:
class Solution:
def combine(self, n: int, k: int) -> List[List[int]]:
res=[]
def DNS(num,j,tmd):
if j==0:
res.append(tmd)
return
for i in range(j,num):
if tmd and i >= tmd[-1]:
continue
else:
DNS(num-1,j-1,tmd+[i])
DNS(n+1,k,[])
return res