有时候需要生成一些列表的全排列组合,例如,有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
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)],