solved(python): programmers

- 연습문제 / 호텔 대실

Author: himitery

programmers/python/연습문제/호텔_대실/__init__.py

@@ -0,0 +1,31 @@
+from typing import List
+
+
+def solution(book_time: List[List[str]]) -> int:
+    times, rooms = (
+        sorted(
+            [(time_to_num(start), time_to_num(end) + 10) for start, end in book_time],
+            key=lambda time: time[0],
+        ),
+        [],
+    )
+
+    for start, end in times:
+        allocated = False
+
+        for idx, room_end in enumerate(rooms):
+            if room_end <= start:
+                rooms[idx] = end
+                allocated = True
+                break
+
+        if not allocated:
+            rooms.append(end)
+
+    return len(rooms)
+
+
+def time_to_num(time: str) -> int:
+    hour, minute = map(int, time.split(":"))
+
+    return hour * 60 + minute

programmers/python/연습문제/호텔_대실/main.py

@@ -0,0 +1,63 @@
+[
+  {
+    "params": [
+      [
+        [
+          "15:00",
+          "17:00"
+        ],
+        [
+          "16:40",
+          "18:20"
+        ],
+        [
+          "14:20",
+          "15:20"
+        ],
+        [
+          "14:10",
+          "19:20"
+        ],
+        [
+          "18:20",
+          "21:20"
+        ]
+      ]
+    ],
+    "expected": 3
+  },
+  {
+    "params": [
+      [
+        [
+          "09:10",
+          "10:10"
+        ],
+        [
+          "10:20",
+          "12:20"
+        ]
+      ]
+    ],
+    "expected": 1
+  },
+  {
+    "params": [
+      [
+        [
+          "10:20",
+          "12:30"
+        ],
+        [
+          "10:20",
+          "12:30"
+        ],
+        [
+          "10:20",
+          "12:30"
+        ]
+      ]
+    ],
+    "expected": 3
+  }
+]

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