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

 

[ Contents ]

 

 

1. 문제 (링크 참조)

15656번: N과 M (7)

 

 

 

2. 문제 풀이

 주어진 수열에서 M개를 뽑는 중복순열을 구하는 문제입니다. 중복순열은 itertools.product() 함수를 이용해서 쉽게 구할 수 있습니다.

 

 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 = product(sorted(numlist), repeat=M)

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

 사전순으로 출력해야 하므로, 입력받은 수열을 오름차순으로 정렬해서 인수에 대입합니다.