[백준] 15649 - N과 M (1) (파이썬, Python)

[문제 출처]

www.acmicpc.net/problem/15649

15649번: N과 M (1)

---

기본적인 순열을 구현하는 문제이다.

 

여러 문제를 풀어 오면서 순열을 이용하는 문제를 여러 번 만났지만 직접 순열을 구현한 적은 없었다.

(python은 itertools 라이브러리를 이용하면 편리하게 순열을 계산할 수 있다.)

 

이 문제를 만나 처음으로 순열을 구현해봤는데 쉽지 않았다.

다른 자료를 참고해서 공부한 후 구현한 코드는 아래와 같다.

---

전체 코드

import sys

def permutation(data, n):
    res = []
    
    if n == 1:
        res = [[e] for e in data]

    elif n >= 2:
        for t in data:
            excluded = data[:]
            excluded.remove(t)

            excludedPermutation = permutation(excluded, n-1)

            for case in excludedPermutation:
                res.append([t]+case)

    return res
    
N, M = map(int, sys.stdin.readline().split(" "))

results = permutation(list(range(1, N+1)), M)

for res in results:
    print(" ".join(map(str, res)))