티스토리 뷰

반응형

문제

 

15652번: N과 M (4)

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

www.acmicpc.net

 

풀이
중복조합 nHr을 구하기 위해서는 itertools의 combinations_with_replacement를 사용하면 된다.

import sys
from itertools import combinations_with_replacement

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

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

반응형