有时候,我们需要列举出所有的元素组合情况。比如从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)
结果: