嵌套排列生成到循环排列生成的算法

来源:互联网 发布:怎么下载气象数据 编辑:程序博客网 时间:2024/06/05 16:20
有时候需要生成一些列表的全排列组合,例如,有3个箱子,第一个中有红、蓝色球各一,第二个有红、蓝、绿色球各一,第三个有红、蓝、绿色球各一,现在在这个3个箱子中任取一个,共有C(1,2) * C(1,3)*C(1,3) = 2 * 3 * 3 = 18 种不同的取法,现在需要列出所有的取法,简单的做法是做个3层嵌套的循环,但当嵌套层数运行时定义时,这样的代码就不太好写,以下的算法将上述的循环转换到一个2维循环中。

A = ['A', 'B']
B = ['A', 'B', 'C']
C = ['A', 'B', 'C']

for a in A:
    for b in B:
        for c in C:
            print a,b,c
   
print '*' * 40

L = [A, B, C]
N = 1
for l in L:
    N *= len(l)

for i in range(0, N):
    step = N
    for l in L:
        step /= len(l)
        print l[i / step % len(l)],
    print
原创粉丝点击