[Brute Force] 백준 15665 N과 M (11) - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

15665번: N과 M (11)

 

 

 

2. 문제 풀이

 주어진 수열에서 M개를 뽑을 중복순열을 구하는 문제입니다.

 

itertools.product(lst, repeat=n): lst에서 n개를 중복해서 뽑을 순열

 

 

 

3. 코드

from itertools import product

N, M = map(int, input().split())
numlist = list(map(int, input().split()))
case = sorted(set(product(numlist, repeat=M)))

for i in case:
    for j in i:
        print(j, end=" ")
    print()

 itertools 라이브러리의 product() 함수를 이용해서 중복 순열을 구합니다. 중복된 순열은 set으로 제거하고, 오름차순으로 출력합니다.