알고리즘 - Python / 백준 - 15650번 : N과 M (2)

15650번: N과 M (2) (acmicpc.net)

15650번: N과 M (2)

 

---

풀이

• 1 ~ N 까지 수 중에서 중복없이 M개씩 고른 오름차순 조합을 사전순 증가로 출력하는 문제이다.
• 파이썬의 itertools 에서 combinations 을 import 하여 쉽게 조합을 구현 할 수 있다.
• 파이썬의 combinations 으로 만든 조합들은 처음부터 사전순 증가와 오름차순으로 만들어지므로 출력만 하면 된다.

 

import sys
from itertools import combinations
n , m = map(int, sys.stdin.readline().split(" "))
stack = []
for i in range(1,n+1):
    stack.append(i)
ans = list(combinations(stack, m))
if m == 1:
    for t in range(0, len(ans)):
        s = list(ans[t])
        print(s[0])
else:
    for t in range(0,len(ans)):
        s = list(ans[t])
        for u in range(0,len(s)-1):
            print(s[u],end=" ")
        print(s[len(s)-1])

---