solved(python): programmers

- 연습문제 / 리코쳇 로봇

Author: himitery

programmers/연습문제/python/리코쳇_로봇/__init__.py

@@ -0,0 +1,43 @@
+from collections import deque
+from typing import List, Tuple
+
+
+def solution(board: List[str]) -> int:
+    matrix = [list(row) for row in board]
+    n, m = len(matrix), len(matrix[0])
+
+    red, green = find_points(matrix)
+    queue, visited = deque([(0, red)]), set()
+
+    while queue:
+        step, (x, y) = queue.popleft()
+        if (x, y) == green:
+            return step
+
+        if (x, y) in visited:
+            continue
+
+        visited.add((x, y))
+
+        for dx, dy in [(0, -1), (0, 1), (-1, 0), (1, 0)]:
+            nx, ny = x, y
+            while 0 <= nx + dx < m and 0 <= ny + dy < n and matrix[ny + dy][nx + dx] != "D":
+                nx, ny = nx + dx, ny + dy
+
+            if (x, y) != (nx, ny):
+                queue.append((step + 1, (nx, ny)))
+
+    return -1
+
+
+def find_points(matrix: List[List[str]]) -> Tuple[Tuple[int, int], Tuple[int, int]]:
+    red, green = (-1, -1), (-1, -1)
+
+    for y, row in enumerate(matrix):
+        for x, value in enumerate(row):
+            if value == "R":
+                red = (x, y)
+            if value == "G":
+                green = (x, y)
+
+    return red, green

programmers/연습문제/python/리코쳇_로봇/main.py

@@ -0,0 +1,25 @@
+[
+  {
+    "params": [
+      [
+        "...D..R",
+        ".D.G...",
+        "....D.D",
+        "D....D.",
+        "..D...."
+      ]
+    ],
+    "expected": 7
+  },
+  {
+    "params": [
+      [
+        ".D.R",
+        "....",
+        ".G..",
+        "...D"
+      ]
+    ],
+    "expected": -1
+  }
+]

programmers/연습문제/python/리코쳇_로봇/sample.json

@@ -0,0 +1,28 @@
+import json
+import os
+import unittest
+
+from parameterized import parameterized
+
+from .main import solution
+
+
+def load_sample(filename: str):
+    path = os.path.join(os.path.dirname(os.path.abspath(__file__)), filename)
+
+    with open(path, "r") as file:
+        return [(case["params"], case["expected"]) for case in json.load(file)]
+
+
+class TestCase(unittest.TestCase):
+    @parameterized.expand(load_sample("sample.json"))
+    def test_case(self, params: list, expected: any):
+        # When
+        result = solution(*params)
+
+        # Then
+        self.assertEqual(expected, result)
+
+
+if __name__ == "__main__":
+    unittest.main()