[파이썬] 순열과 조합 - permutations, combinations

알고리즘 문제를 풀 때, 순열과 조합 개념이 필요한 문제들이 있다.

 

itertools 라이브러리를 사용하면 매우 간단하게 순열과 조합을 만들 수 있다.

---

itertools

파이썬에 있는 표준 라이브러리다.

다양한 iterator가 있지만 그중 permutations(), combinations()만 간단하게 다룬다.

 

---

permutations()

from itertools import permutations

arr = [1, 2, 3, 4]
for p in permutations(arr, 2):
    print(p)

결과:

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

 

tuple 형태로 나온다.

 

---

combinations()

from itertools import combinations

arr = [1, 2, 3, 4]
for c in combinations(arr,3):
    print(c)

결과:

(1, 2, 3)

(1, 2, 4)

(1, 3, 4)

(2, 3, 4)