[백준15657] N과 M (8)

땅어 2020. 3. 14. 15:35

728x90

문제

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

15657번: N과 M (8)

N개의 자연수와 자연수 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오. N개의 자연수는 모두 다른 수이다. N개의 자연수 중에서 M개를 고른 수열 같은 수를 여러 번 골라도 된다. 고른 수열은 비내림차순이어야 한다. 길이가 K인 수열 A가 A1 ≤ A2 ≤ ... ≤ AK-1 ≤ AK를 만족하면, 비내림차순이라고 한다.

www.acmicpc.net

생각의 흐름

N과 M (7) 문제와 유사하나 순열이 아닌 조합을 구하는 문제

코드

from itertools import combinations_with_replacement

N,M = map(int, input().split())
arr = list(map(int, input().split()))
arr.sort()

for case in combinations_with_replacement(arr, M):
    print(' '.join(map(str, case)))

728x90