티스토리 뷰

PS/Python

[Python] 백준 알고리즘 #15651 N과 M(3) - 중복 순열

yoo.o 2020. 8. 12. 02:57
반응형

문제

 

15651번: N과 M (3)

한 줄에 하나씩 문제의 조건을 만족하는 수열을 출력한다. 중복되는 수열을 여러 번 출력하면 안되며, 각 수열은 공백으로 구분해서 출력해야 한다. 수열은 사전 순으로 증가하는 순서로 출력해

www.acmicpc.net

 

풀이

중복을 허용하는 순열을 구하기 위해서는 itertools의 product를 사용해야한다.

product(리스트, repeat=n)

import sys
from itertools import product

N, M = map(int, sys.stdin.readline().split(" "))
li = []
for i in range(1,N+1):
    li.append(i)

for each in list(product(li, repeat=M)):
    for num in each:
        print(num, end=" ")
    print()

반응형
저작자표시