solved(python): programmers

- 2023 현대모비스 알고리즘 경진대회 예선 / 에어컨

Author: himitery

programmers/python/2023_현대모비스_알고리즘_경진대회_예선/에어컨/__init__.py

@@ -0,0 +1,26 @@
+import math
+
+from typing import List
+
+
+def solution(temperature: int, t1: int, t2: int, a: int, b: int, onboard: List[int]) -> int:
+    dp = [[math.inf for _ in range(-11, 42)] for _ in range(len(onboard))]
+    dp[0][temperature] = 0
+
+    for time, customer in enumerate(onboard):
+        for temp in range(-10, 41):
+            if customer and not t1 <= temp <= t2:
+                dp[time][temp] = math.inf
+                continue
+
+            dp[time][temp] = min(
+                dp[time][temp],
+                dp[time - 1][temp] + b,
+                dp[time - 1][temp - 1] + a,
+                dp[time - 1][temp + 1] + a,
+                dp[time - 1][temp] if temp == temperature else math.inf,
+                dp[time - 1][temp - 1] if temp - 1 < temperature else math.inf,
+                dp[time - 1][temp + 1] if temp + 1 > temperature else math.inf,
+            )
+
+    return int(min(dp[-1]))

programmers/python/2023_현대모비스_알고리즘_경진대회_예선/에어컨/main.py

@@ -0,0 +1,88 @@
+[
+  {
+    "params": [
+      28,
+      18,
+      26,
+      10,
+      8,
+      [
+        0,
+        0,
+        1,
+        1,
+        1,
+        1,
+        1
+      ]
+    ],
+    "expected": 40
+  },
+  {
+    "params": [
+      -10,
+      -5,
+      5,
+      5,
+      1,
+      [
+        0,
+        0,
+        0,
+        0,
+        0,
+        1,
+        0
+      ]
+    ],
+    "expected": 25
+  },
+  {
+    "params": [
+      11,
+      8,
+      10,
+      10,
+      1,
+      [
+        0,
+        1,
+        1,
+        1,
+        1,
+        1,
+        1,
+        0,
+        0,
+        0,
+        1,
+        1
+      ]
+    ],
+    "expected": 20
+  },
+  {
+    "params": [
+      11,
+      8,
+      10,
+      10,
+      100,
+      [
+        0,
+        1,
+        1,
+        1,
+        1,
+        1,
+        1,
+        0,
+        0,
+        0,
+        1,
+        1
+      ]
+    ],
+    "expected": 60
+  }
+]

programmers/python/2023_현대모비스_알고리즘_경진대회_예선/에어컨/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()