[백준]N과 M (2)

이번에는 자연수 n과 m이 주어질때 길이가 m인 수열을 구하는 알고리즘을 만들어봅니다.

import itertools

N, M = map(int, input().split())
num_list = [i for i in range(1, N+1)]
    
for num in itertools.combinations(num_list, M):
    for i in num:
        print(i, end = ' ')
    print(end = '\n')

처음에는 itertool모듈을 가져와줍니다 다음 n과 m의 입력값을 만들어주고 리스트하나를 만들어줍니다 다음 순열과 조합을 구하는 itertool.permutations를 사용하여 반복시켜주고 그안에 for문을 한번더 사용하여 배열을 만들수 있도록 만들어줍니다 마지막으로 출력해줍니다