[Brute Force] 백준 15663 N과 M (9) - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

15663번: N과 M (9)

 

 

 

2. 문제 풀이

 주어진 수열에서 M개를 뽑는 순열을 구하는 문제입니다. 다만, 중복되는 순열을 여러 번 출력해선 안됩니다.

 

itertools.permutations(lst, n): lst에서 n개를 뽑는 순열

 위 함수를 사용하면 쉽게 순열을 구할 수 있습니다. 주어진 수열에서 순열을 구하고, 중복되는 순열만 set으로 제거해줍니다.

 

 

 

3. 코드

from itertools import permutations

N, M = map(int, input().split())
numlist = list(map(int, input().split()))
case = sorted(set(permutations(numlist, M)))

for i in case:
    for j in i:
        print(j, end=" ")
    print()

 중복되는 순열을 set으로 제거하고, 오름차순으로 정렬해서 출력합니다.