[백준15652] N과 M (4)

문제

https://www.acmicpc.net/problem/15652

15652번: N과 M (4)

 

생각의 흐름

---

이전에 포스팅 했던 N과 M (3)과 비교했을 때 중복선택을 허락하는 것은 동일하지만 순열이 아닌 조합이라는 점이 다르다.

 

깨달은 점

---

파이썬 itertools에는 이를 지원하는, 즉 중복선택을 허락하는 조합을 구하는 combinations_with_replacement라는 함수가 있다.

 

코드

---

from itertools import combinations_with_replacement

N,M = map(int, input().split())
arr = list(range(1,N+1))
# 중복을 허용하는 조합
for case in combinations_with_replacement(arr, M):
    print(' '.join(map(str, case)))