백준 15654번 N과 M (5)

문제 링크: https://www.acmicpc.net/problem/15654

Author: skysign

백준 15654번 N과 M (5)/main.py

@@ -0,0 +1,37 @@
+import sys
+from typing import List
+
+
+def solve():
+    N, M = map(int, sys.stdin.readline().strip().split(' '))
+    dt: List[int] = list(map(int, sys.stdin.readline().strip().split(' ')))
+    dt.sort()
+
+    visited: List[bool] = [False for _ in range(len(dt))]
+    my_perm(visited, dt, [], M, perm_found)
+
+
+def perm_found(perms: List[int]):
+    perms = list(map(lambda x: str(x) + ' ', perms))
+    perm = ''.join(perms).strip()
+    print(perm)
+
+
+def my_perm(visited: List[bool], dt, perms, length, perm_found):
+    if len(perms) == length:
+        perm_found(perms)
+        return
+
+    for i in range(len(dt)):
+        if visited[i]:
+            continue
+
+        visited[i] = True
+        perms.append(dt[i])
+        my_perm(visited, dt, perms, length, perm_found)
+        perms.pop()
+        visited[i] = False
+
+
+if __name__ == '__main__':
+    solve()

백준 15654번 N과 M (5)/test1.txt

@@ -0,0 +1,2 @@
+3 1
+4 5 2

백준 15654번 N과 M (5)/test1_answer.txt

@@ -0,0 +1,3 @@
+2
+4
+5

백준 15654번 N과 M (5)/test2.txt

@@ -0,0 +1,2 @@
+4 2
+9 8 7 1

백준 15654번 N과 M (5)/test2_answer.txt

@@ -0,0 +1,12 @@
+1 7
+1 8
+1 9
+7 1
+7 8
+7 9
+8 1
+8 7
+8 9
+9 1
+9 7
+9 8

백준 15654번 N과 M (5)/test3.txt

@@ -0,0 +1,2 @@
+4 4
+1231 1232 1233 1234

백준 15654번 N과 M (5)/test3_answer.txt

@@ -0,0 +1,24 @@
+1231 1232 1233 1234
+1231 1232 1234 1233
+1231 1233 1232 1234
+1231 1233 1234 1232
+1231 1234 1232 1233
+1231 1234 1233 1232
+1232 1231 1233 1234
+1232 1231 1234 1233
+1232 1233 1231 1234
+1232 1233 1234 1231
+1232 1234 1231 1233
+1232 1234 1233 1231
+1233 1231 1232 1234
+1233 1231 1234 1232
+1233 1232 1231 1234
+1233 1232 1234 1231
+1233 1234 1231 1232
+1233 1234 1232 1231
+1234 1231 1232 1233
+1234 1231 1233 1232
+1234 1232 1231 1233
+1234 1232 1233 1231
+1234 1233 1231 1232
+1234 1233 1232 1231

백준 15654번 N과 M (5)/test_main.py

@@ -0,0 +1,33 @@
+import sys
+from pathlib import Path
+from unittest import TestCase
+from main import solve
+
+
+def my_solve(testcase_input):
+    sys.stdin = open(testcase_input, 'r')
+    stdout = sys.stdout
+    sys.stdout = open('stdout.txt', 'w')
+    solve()
+    sys.stdout.close()
+    sys.stdout = stdout
+
+
+class Test(TestCase):
+    def test_solve(self, testcase_number: str):
+        my_solve('test' + testcase_number + '.txt')
+        self.assertEqual(
+            Path('test' + testcase_number + '_answer.txt').read_text().strip(),
+            Path('stdout.txt').read_text().strip())
+
+    def test1_solve(self):
+        self.test_solve('1')
+
+    def test2_solve(self):
+        self.test_solve('2')
+
+    def test3_solve(self):
+        self.test_solve('3')
+
+    def test4_solve(self):
+        self.test_solve('4')