티스토리 뷰

PS/Python

[Python] 백준 알고리즘 #15650 N과 M(2) - 조합

yoo.o 2020. 8. 12. 02:52
반응형

문제

 

15650번: N과 M (2)

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

www.acmicpc.net

 

풀이

오름차순으로 정렬해야한다는 점에서 N과 M(1) 문제와 차이가 있다. 

즉, 오름차순이 아닌 순열 (예. 1 2 4 3)은 다 프린트하지 말아야한다.

따라서 조합으로 풀었다.

import sys
from itertools import combinations

N, M = map(int, sys.stdin.readline().split(" "))
li = []
for i in range(1,N+1):
    li.append(i)

for each in list(combinations(li, M)):
    for num in each:
        print(num, end=" ")
    print()

반응형
저작자표시