一、问题描述
在数学上,我们经常遇到这样的问题:有编号分别为1,2,3,4,5的5个小球,从中选出3个,有多少种组合?利用数学公式,我们可以很方便地算出组合的数量。那么如何通过编程来枚举这些组合呢?也就是说,我们希望得到[1,2,3],[2,3,4]……这样的结果。当然,已经有现成的API能够帮助我们实现这个需求。接下来,我们就看看如何利用递归,自己写出这样一个用于枚举组合的函数。
二、代码实现
下面实现了两个版本的枚举组合函数。
- 从一个元素不重复的列表中选出n个元素
- 从1到n这n个数字中选出r个
def combinations(source: list,n:int)->list:
'''从一个元素不重复的列表里选出n个元素
:参数 source:元素不重复的列表
:参数 n: 要选出的元素数量,正整数,小于等于列表source的长度
'''
# 如果n正好等于列表的长度,那么只有一种组合
# 就是把所有元素都选出来
if len(source)==n:
return [source]
# 如果n是1,那么列表中的每个元素都是一个组合
if n == 1:
ans = []
for i in source:
ans.append([i])
return ans
# 下面处理n小于列表长度的情况
ans = []
# 从列表里选n个元素出来,可以理解为先把列表里的第0个元素拿出来放进组合
# 然后从剩下的元素里选出n-1个
for each_list in combinations(source[1:],n-1):
ans.append([source[0]]+each_list)
# 还可以直接从剩下的元素里选出n个
for each_list in combinations(source[1:],n):
ans.append(each_list)
return ans
print(combinations(["A",'B','C','D','E','F'],2))
def com(n: int, r: int)-> list:
'''从1到n这个n个数里选出r个
:参数 n:正整数
:参数 r:正整数,不大于n
'''
if n == r:
return [[i for i in range(1,n+1)]]
if r == 1:
return [[i] for i in range(1,n+1)]
ans = []
# 先把n拿出来放进组合,然后从剩下的n-1个数里选出r-1个数
for each_list in com(n-1,r-1):
ans.append([n]+each_list)
# 直接从剩下的n-1个数里,选出r个
for each_list in com(n-1,r):
ans.append(each_list)
return ans
print(com(5,3))