(Python) 백준 15652번 N과 M(4) - class 4

27 Feb 2021 in Algorithm on BOJ

[문제 링크]

15652번 N과 M(4)

---

접근 방법
주어진 수 중에서 중복을 포함해서 m개만큼 뽑는 문제다. itertools 라이브러리에 combinations_with_replacement 중복조합을 생각해 냈다.

해결

1. num 리스트에 1부터 n까지의 자연수를 만들어 놓는다.
2. combinatiocombinations_with_replacementns로 num에서 m개를 뽑는다.
3. combinations num의 앞에서부터 차례로 뽑아내는데 num이 현재 오름차순이므로 오름차순으로 뽑혀 나온다.
4. 출력

from itertools import combinations_with_replacement

n, m = map(int, input().split())
num = [x for x in range(1, n + 1)]

for i in list(combinations_with_replacement(num, m)):
    for j in i:
        print(j, end=" ")
    print()