Python递归实现枚举组合(从n个不重复的元素中选出r个)

一、问题描述

在数学上，我们经常遇到这样的问题：有编号分别为1,2,3,4,5的5个小球，从中选出3个，有多少种组合？利用数学公式，我们可以很方便地算出组合的数量。那么如何通过编程来枚举这些组合呢？也就是说，我们希望得到[1,2,3],[2,3,4]……这样的结果。当然，已经有现成的API能够帮助我们实现这个需求。接下来，我们就看看如何利用递归，自己写出这样一个用于枚举组合的函数。

二、代码实现

下面实现了两个版本的枚举组合函数。

1. 从一个元素不重复的列表中选出n个元素
2. 从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))

 

如果这篇博文帮到了你，就请给我点个赞吧(#^.^#)

有疑问也欢迎留言~博主可nice啦，在线秒回ヾ(◍°∇°◍)ﾉﾞ