코딩 테스트/백준

[백준15664] N과 M (10)

땅어 2020. 3. 14. 15:53
728x90
300x250

문제

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

 

15664번: N과 M (10)

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

www.acmicpc.net

 

생각의 흐름

N과 M (9)와 유사하지만 조합을 구한다는 점이 다르다.

 

코드

from itertools import combinations

N,M = map(int, input().split())
arr = list(map(int, input().split()))
arr.sort()
result = []
result = set(result)
for case in combinations(arr, M):
    if case not in result:
        result.add(case)
        print(' '.join(map(str, case)))

 

728x90
300x250