[백준] 15650번: N과 M(2)

문제

백준 15650
자연수 N과 M이 주어졌을 때, 아래 조건을 만족하는 길이가 M인 수열을 모두 구하는 프로그램을 작성하시오.

1부터 N까지 자연수 중에서 중복 없이 M개를 고른 수열
고른 수열은 오름차순이어야 한다.

어떻게 풀면 좋을까?

음 .. 사실 보자마자 떠오른건 조합... 이었다.
파이썬의 정말 좋은 점이다.

풀이

from itertools import combinations

n, m = map(int, input().split())
# 1부터 저장하기 위해서 1을 더해줬다.
lst = [i + 1 for i in range(n)]

for i in combinations(lst, m):
# 문제에서 요구하는대로 출력하기 위해서 print문을 다음과 같이 출력했다.
    for j in i:
        print(j, end=" ")
    print()