経営工学徒の備忘録

~日々の大学生活や留学について~

pythonで順列全探索を実装する

以前、bit全探索に関する記事をあげました。
今回は、pythonで順列全探索を実装してみます。

itertools.permutationsというライブラリを利用すれば、非常に簡単に実装できるようです。

from itertools import permutations
nums =[1,2,3]
pers =permutations(nums,3)
for i in pers:
    print(i)

permutationsがイテレータであるということに注意が必要です。
二個目の引数に出力したい数を入れてfor文を回してあげます。

出力結果

(1, 2, 3)
(1, 3, 2)
(2, 1, 3)
(2, 3, 1)
(3, 1, 2)
(3, 2, 1)