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

 

[ Contents ]

 

 

1. 문제 (링크 참조)

15651번: N과 M (3)

 

 

 

2. 문제 풀이

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

 

 

 

3. 코드

from itertools import product

N, M = map(int, input().split())
case = product(range(1, N+1), repeat=M)

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

 product() 함수를 이용해서 수열에서 M개를 뽑은 중복 순열을 구합니다.

 

 

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