[백준풀때 유용한 파이썬 팁] itertools을 이용해 조합(combinations)과 순열(permutations), 중복순열(product) 사용하기

# 발단

11050번: 이항 계수 1

 이항계수문제이다. dp나 재귀를 이용해서 풀  수 있을 테지만, 파이썬 itertools는 조합과 순열기능을 지원해준다. 

 

# 사용법

from itertools import combinations, permutations


arr = ["A","B","C"]

# 조합
combi = combinations(arr, 2)

# 순열
permu = permutations(arr,2)

(리스트, 뽑을 개수)와 같이 지정해준다면 쉽게 사용할 수 있다.

 

## 출력

출력을 하고 싶다면 list로 이터레이터를 변환시켜준다.

print(f"조합 : {list(combi)}")
print(f"순열 : {list(permu)}")

 

조합의 개수를 구하고 싶다면, 해당 리스트의 len을 구하면 된다.

print(len(list(combi)))

 

# 번외

중복순열은 product를 이용하자!

from itertools import product
product_ = product(range(n), k)

 

# 주의할점

다만 itertools은 시간복잡도가 꽤 크기때문에 사용에 주의해야하고, 일부 코딩테스트에서는 itertools를 사용하지 못하게 하기도 한다고 한다.