这个问题可以使用递归的方式来解决。下面是一个示例代码:
def find_subsets(N, K):
subsets = []
helper(N, K, 1, [], subsets)
return subsets
def helper(N, K, start, subset, subsets):
if len(subset) == K: # 找到一个符合条件的子集
subsets.append(subset[:]) # 注意要复制一份subset加入到subsets中
return
for i in range(start, N+1):
subset.append(i) # 添加当前元素到子集中
helper(N, K, i+1, subset, subsets) # 递归调用helper,继续添加下一个元素
subset.pop() # 回溯,将最后添加的元素移除
# 测试示例
N = 5
K = 3
subsets = find_subsets(N, K)
for subset in subsets:
print(subset)
运行以上代码输出:
[1, 2, 3]
[1, 2, 4]
[1, 2, 5]
[1, 3, 4]
[1, 3, 5]
[1, 4, 5]
[2, 3, 4]
[2, 3, 5]
[2, 4, 5]
[3, 4, 5]
这个代码使用了递归函数helper来辅助查找所有符合条件的子集。helper函数接受四个参数:N表示元素的范围(1到N),K表示子集的长度,start表示当前要添加的元素的起始值,subset表示当前已经形成的子集,subsets表示保存所有符合条件的子集的列表。
在helper函数中,首先检查当前子集的长度是否等于K,如果是,则将子集加入到subsets中,并返回。否则,从start开始循环遍历元素,将元素添加到子集中,然后递归调用helper函数,继续添加下一个元素。在递归调用之后,需要回溯,将最后添加的元素移除,以便尝试添加其他元素。
最后,调用find_subsets函数来查找所有符合条件的子集,并打印输出。
下一篇:保持与变长代码匹配的记录