[백준] 15654 - N과 M (5) (백트래킹, permutations)

v

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

15654번: N과 M (5)

from itertools import permutations
 
n, m = map(int, input().split())
arr = list(map(int, input().split()))
arr.sort()
 
# for perm in permutations(arr, m):
#     for i in perm:
#         print(i, end=' ')
#     print()
 
 
def backtracking(li):
    if len(li) == m:
        for i in li:
            print(i, end=' ')
        print()
    elif len(li) > len(arr):
        return
 
    for i in range(0, n):
        if arr[i] not in li:
            backtracking(li + [arr[i]])
 
 
backtracking(list())

중복 없는 순열을 구하는 문제이다.

파이썬으로는 permutations를 이용해서 간단히 구현할 수 있고(주석), 세부 로직은 아래 코드와 같다.