알고리즘 - Python / 백준 - 15654번 : N과 M (5)

15654번: N과 M (5) (acmicpc.net)

15654번: N과 M (5)

 

---

풀이

• 중복되지 않은 N가지 수를 입력 받고 , 그 중 M개를 뽑아 순열을 만들고 사전순으로 출력하면 되는 문제이다.
• 파이썬의 itertools 에서 permutations 을 import 하여 쉽게 순열을 구현 할 수 있다.
• 입력받은 N개의 수를 sort해 permutations 의 결과가 사전순으로 나올 수 있게 한다.

 

import sys
from itertools import permutations
n , m = map(int, sys.stdin.readline().split(" "))
stack = list(map(int , sys.stdin.readline().split(" ")))
stack.sort()
ans = list(permutations(stack, m))
if m == 1:
    for t in range(0, len(ans)):
        s = list(ans[t])
        print(s[0])
else:
    for t in range(0,len(ans)):
        s = list(ans[t])
        for u in range(0,len(s)-1):
            print(s[u],end=" ")
        print(s[len(s)-1])

---