백준 1697번 숨바꼭질

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

Author: skysign

백준 1697번 숨바꼭질/main.py

@@ -0,0 +1,49 @@
+import sys
+from typing import List
+
+# python3로는 시간초과로 정답 판정이 안되고
+# pypy3로 해서 실행시간을 좀 줄여야, 정답 인정이 됩니다.
+def solve():
+    N, K = map(int, sys.stdin.readline().strip().split(' '))
+    mx = max(N, K)
+    dt: List[int] = [sys.maxsize for _ in range(mx + 10)]
+    l = len(dt)
+    dt[N] = 0
+
+    if N == K:
+        print(0)
+        return
+
+    time = 0
+    queue = []
+    add_plus_minus_x2(queue, l, dt, N, time)
+
+    while dt[K] == sys.maxsize:
+        time += 1
+        queue2 = []
+
+        for n in queue:
+            dt[n] = min(dt[n], time)
+
+            if n == K:
+                print(time)
+                return
+
+            add_plus_minus_x2(queue2, l, dt, n, time)
+
+        queue = queue2
+
+    print(time)
+
+
+def add_plus_minus_x2(queue: List[int], l, dt, N, time):
+    if (N - 1 >= 0) and (N - 1 not in queue) and (dt[N - 1] > time):
+        queue.append(N - 1)
+    if (N + 1 < l) and (N + 1 not in queue) and (dt[N + 1] > time):
+        queue.append(N + 1)
+    if (N * 2 < l) and (N * 2 not in queue) and (dt[N * 2] > time):
+        queue.append(N * 2)
+
+
+if __name__ == '__main__':
+    solve()

백준 1697번 숨바꼭질/test1.txt

@@ -0,0 +1 @@
+5 17

백준 1697번 숨바꼭질/test1_answer.txt

@@ -0,0 +1 @@
+4

백준 1697번 숨바꼭질/test_main.py

@@ -0,0 +1,23 @@
+import sys
+from pathlib import Path
+from unittest import TestCase
+from main import solve
+
+
+class Test(TestCase):
+    def my_solve(self, testcase_input):
+        sys.stdin = open(testcase_input, 'r')
+        stdout = sys.stdout
+        sys.stdout = open('stdout.txt', 'w')
+        solve()
+        sys.stdout.close()
+        sys.stdout = stdout
+
+    def test_solve(self, testcase_number: str):
+        self.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')