solved(python): programmers

- PCCP 기출문제 / 충돌위험 찾기

Author: himitery

programmers/PCCP_기출문제/python/충돌위험_찾기/__init__.py

@@ -0,0 +1,30 @@
+from collections import Counter
+from typing import List
+
+
+def solution(points: List[List[int]], routes: List[List[int]]) -> int:
+    points = [(y - 1, x - 1) for y, x in points]
+    queue, count = [(0, points[route[0] - 1], [r - 1 for r in route]) for route in routes], 0
+
+    while queue:
+        new_queue, counter = [], Counter([point for idx, point, path in queue])
+        count += len([key for key, count in counter.items() if count > 1])
+
+        for idx, (y, x), path in queue:
+            if (y, x) == points[path[idx]]:
+                idx += 1
+
+            if idx == len(path):
+                continue
+
+            target_y, target_x = points[path[idx]]
+            if y != target_y:
+                dy = 1 if y < target_y else -1
+                new_queue.append((idx, (y + dy, x), path))
+            else:
+                dx = 1 if x < target_x else -1
+                new_queue.append((idx, (y, x + dx), path))
+
+        queue = new_queue
+
+    return count

programmers/PCCP_기출문제/python/충돌위험_찾기/main.py

@@ -0,0 +1,51 @@
+[
+  {
+    "params": [
+      [
+        [3, 2],
+        [6, 4],
+        [4, 7],
+        [1, 4]
+      ],
+      [
+        [4, 2],
+        [1, 3],
+        [2, 4]
+      ]
+    ],
+    "expected": "1"
+  },
+  {
+    "params": [
+      [
+        [3, 2],
+        [6, 4],
+        [4, 7],
+        [1, 4]
+      ],
+      [
+        [4, 2],
+        [1, 3],
+        [4, 2],
+        [4, 3]
+      ]
+    ],
+    "expected": "9"
+  },
+  {
+    "params": [
+      [
+        [2, 2],
+        [2, 3],
+        [2, 7],
+        [6, 6],
+        [5, 2]
+      ],
+      [
+        [2, 3, 4, 5],
+        [1, 3, 4, 5]
+      ]
+    ],
+    "expected": "0"
+  }
+]

programmers/PCCP_기출문제/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: str):
+        # When
+        result = solution(*params)
+
+        # Then
+        self.assertEqual(expected, str(result))
+
+
+if __name__ == "__main__":
+    unittest.main()