solved(python): programmers

- PCCP 기출문제 / 광물 캐기

Author: himitery

programmers/연습문제/python/광물_캐기/__init__.py

@@ -0,0 +1,33 @@
+import math
+
+from collections import deque
+
+
+def solution(picks, minerals):
+    n = len(minerals)
+    stack, minimum = deque([(0, 0, picks)]), math.inf
+
+    mineral_cost = {"diamond": 25, "iron": 5, "stone": 1}
+    while stack:
+        idx, cost, remains = stack.pop()
+        if cost > minimum:
+            continue
+
+        if idx == (n // 5 + int(n % 5 != 0)) or sum(remains) == 0:
+            minimum = min(minimum, cost)
+            continue
+
+        for pick, remain in enumerate(remains):
+            if remain == 0:
+                continue
+
+            new_cost = 0
+            for mineral in minerals[idx * 5 : min((idx + 1) * 5, n)]:
+                new_cost += max(mineral_cost[mineral] // 5 ** (2 - pick), 1)
+
+            new_remains = remains[:]
+            new_remains[pick] -= 1
+
+            stack.append((idx + 1, cost + new_cost, new_remains))
+
+    return minimum

programmers/연습문제/python/광물_캐기/main.py

@@ -0,0 +1,45 @@
+[
+  {
+    "params": [
+      [
+        1,
+        3,
+        2
+      ],
+      [
+        "diamond",
+        "diamond",
+        "diamond",
+        "iron",
+        "iron",
+        "diamond",
+        "iron",
+        "stone"
+      ]
+    ],
+    "expected": 12
+  },
+  {
+    "params": [
+      [
+        0,
+        1,
+        1
+      ],
+      [
+        "diamond",
+        "diamond",
+        "diamond",
+        "diamond",
+        "diamond",
+        "iron",
+        "iron",
+        "iron",
+        "iron",
+        "iron",
+        "diamond"
+      ]
+    ],
+    "expected": 50
+  }
+]

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()