항해99 리부트코스 알고리즘 1주 5일차 15650 N과 M (2)

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

 

문제

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

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

입력

첫째 줄에 자연수 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)

출력

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

수열은 사전 순으로 증가하는 순서로 출력해야 한다.

 

풀이

 

from itertools import combinations
import sys
input = sys.stdin.read
def solve_problem(n, m):
    comb = combinations(range(1, n + 1), m)
    comb_list = []
    for c in comb:
        comb_list.append(' '.join(map(str, c)))
    return comb_list

if __name__ == "__main__":
  
    data = input().split()
    n = int(data[0])
    m = int(data[1])
    
    results = solve_problem(n, m)

    for result in results:
        print(result)