[백준] 15649 N과 M(1) Python

Algorithm (PS)

[백준] 15649 N과 M(1) Python

minjiwoo 2022. 10. 6. 22:19

728x90

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

15649번: N과 M (1)

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

www.acmicpc.net

dfs로도 풀수있다고 하는데, 문제보자마자 난 파이썬 permutation 모듈이 생각나서 그냥 그렇게 풀었는데
메모리초과도 안나고 시간초과도 안나서 통과했다

from itertools import permutations

n, m = map(int, input().split())
array = [i for i in range(1, n+1)] # 1...n
result = []
for i in list(permutations(array, m)):
    result.append(list(i))

result.sort()
for i in result:
    print(" ".join(map(str, i)))

728x90