python全排列算法_Python itertools 模块中的全排列算法，看起来简单却非常让人费解!...

在刷 leetcode 时遇到了全排列问题，想来想去也只能想出递归的解法。然后突然想起 Python 的 itertools 模块中有全排列的函数，模块源码是用 C 写的，不过在官方文档中有提供 Python 版本的代码，代码看起来非常简单，但是我看了很久都看不懂算法的原理是什么。

接着在 SO 上发现了相关问题，根据“ Alex Martelli ”的回答，该算法涉及到了Cyclic permutation 理论。有兴趣的同学可以研究一下代码和 Cyclic permutation 理论。

附上代码

def permutations(iterable, r=None):

# permutations('ABCD', 2) --> AB AC AD BA BC BD CA CB CD DA DB DC

# permutations(range(3)) --> 012 021 102 120 201 210

pool = tuple(iterable)

n = len(pool)

r = n if r is None else r

if r > n:

return

indices = list(range(n))

cycles = list(range(n, n-r, -1))

yield tuple(pool[i] for i in indices[:r])

while n:

for i in reversed(range(r)):

cycles[i] -= 1

if cycles[i] == 0:

indices[i:] = indices[i+1:] + indices[i:i+1]

cycles[i] = n - i

else:

j = cycles[i]

indices[i], indices[-j] = indices[-j], indices[i]

yield tuple(pool[i] for i in indices[:r])

break

else:

return