itertools의 순열(permutation), 조합(combination)

from itertools import permutations
from itertools import combinations

mylist = [1, 2, 3]

#순열
print(list(permutations(mylist,  len(mylist))))
>> [[1, 2, 3], [1, 3, 2], [2, 1, 3], [2, 3, 1], [3, 1, 2], [3, 2, 1]]

#조합
print(list(combinations(mylist, 2)))
>> [[1,2],[1,3],[2,3]]

 

코딩 테스트를 하다보면 순열, 조합 문제가 나온다. 

예전 C++으로 문제를 풀던 시절에는 재귀함수를 이용해서 풀었겠지만 이렇게 쉽게 구현되어 있다.