모듈로 순열,조합 구하기(python)(파이썬 잘하고 싶다)

외장함수 itertools안에있는 combinations,permutations를 이용해서 배열의 조합을 구했다.

combinations(조합)

from itertools import permutations
n = int(input()) #n개의 순열
arr = list(map(int,input().split())) # 순열을 구할 배열
li = permutations(arr,n) #permutations(순열을 구할 배열, 구할개수)
for i in li:
    print(i)

 

permutations(순열)

from itertools import combinations
n = int(input()) #n개의 조합
arr = list(map(int,input().split())) # 조합을 구할 배열
li = combinations(arr,n) # combinations(조합을 구할배열, 조합 개수)
for i in li:
    print(i)
    

 

특수한경우 :

중복이 가능한경우

 

from itertools import combinations_with_replacement
n,m = map(int,input().split())
arr = [i+1 for i in range(n)]
for i in combinations_with_replacement(arr,m):
    print(*i)