[백준] N과 M(5) (파이썬)

문제 출처: 백준 온라인 저지

15654번: N과 M (5)

 

문제 설명

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

입력
- 첫째 줄에 N과 M이 주어진다. (1 ≤ M ≤ N ≤ 8)
- 둘째 줄에 N개의 수가 주어진다. 입력으로 주어지는 수는 10,000보다 작거나 같은 자연수이다.

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

---

드디어 브루트포스 1번 카테고리를 끝내고 N, M 시리즈로 넘어왔다.

 

이 전에 문제들을 풀면서 N과 M 문제를 4번까지 끝내놔서 바로 5번으로 넘어왔는데, 1번 카테고리에서 고생한 것과 달리 좀 쉬운 문제가 나와서 한숨을 돌렸다.

 

<정답 코드>

from itertools import permutations

n, m = map(int, input().split())
arr = list(map(int, input().split()))

for p in sorted(permutations(arr, m)):
    print(*p)

접근은 따로 설명하지 않겠다.

이번 문제에서는 순열이라는 언급이 바로 나왔으며, 모든 가능한 경우의 수를 출력하고 N과 M이 8 이하라는 조건이 있었기에 망설임 없이 permutations를 이용했다.

 

ㅇ후에 올라올 N과 M시리즈는 정답률이 50% 미만인 것 들에 대해서만 포스팅 하기로 결정했다.