利用python函数的递归调用实现列举出从n个列表元素取出k个元素的所有组合情况

有时候,我们需要列举出所有的元素组合情况。比如从n个元素选取k个出来组合,根据数学知识可知总共组合数为Cnk。那么具体组合情况如何呢?

代码实现的基本思想就是,先选取一个元素出来,接下来任务就相当于是对后面的元素选取k-1个出来组合。这样就形成了递归调用!

附代码:

# -*- coding: utf-8 -*-

def Combinations(L, k):
    """List all combinations: choose k elements from list L"""
    n = len(L)
    result = [] # To Place Combination result
    for i in range(n-k+1):
        if k > 1:
            newL = L[i+1:]
            Comb, _ = Combinations(newL, k - 1)
            for item in Comb:
                item.insert(0, L[i])
                result.append(item)
        else:
            result.append([L[i]])
    return result, len(result)

if __name__ == '__main__':
    print Combinations(range(5), 3)

结果:

猜你喜欢

转载自blog.csdn.net/qq_27061325/article/details/79905600