[백준 15650] N과 M (2) with Python

문제 링크

www.acmicpc.net/problem/15650

15650번: N과 M (2)

풀이

import sys
import copy

n, m = map(int, sys.stdin.readline().split())

def recursion(n, m, start, seq):
    if len(seq) == m: print(" ".join(map(str, seq)))

    for i in range(start, n+1):
        temp_seq = copy.copy(seq) + [i]
        recursion(n, m, i+1, temp_seq)

recursion(n, m, 1, [])

재귀 함수를 만들어서 숫자를 하나 뽑으면 그것보다 더 큰 수들 중에서 재귀적으로 계속 뽑아서 m개를 뽑을 때마다 수열을 출력하도록 하였다.

 

파이썬 itertools 모듈의 combinations를 사용하면 더 쉽게 해결할 수 있다.

itertools.combinations(iterable, r)는 iterable의 요소들 중에서 r개를 추출하여 중복되지 않게 오름차순 순서로 된 수열을 튜플로 저장하여 반환한다.

import sys
import itertools

n, m = map(int, sys.stdin.readline().split())
comb = itertools.combinations([str(x) for x in range(1, n+1)], m)
print("\n".join(map(" ".join, comb)))