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

 

[ Contents ]

 

 

1. 문제 (링크 참조)

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

15652번: N과 M (4)

 

 

2. 문제 풀이

 길이가 N인 수열에서 중복을 허용해서 M개 뽑는 조합을 구하는 문제입니다.

 

2022.03.31 - [Algorithm] - [Brute Force] 백준 15650 N과 M (2) - 파이썬(Python)

 이전 N과 M(2) 문제와 달리, '중복을 '허용'해서 조합을 구해야 합니다.

 

 

 

3. 코드

from itertools import combinations_with_replacement

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

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

 파이썬에서는 중복 조합을 구하는 함수를 제공합니다. combinations_with_replacement()을 이용해서 중복조합을 구하고 출력합니다.

 

combinations_with_replacement(lst, n): lst에서 중복을 허용해서 n개를 뽑는 조합