solved(python): programmers

- 2024 KAKAO WINTER INTERNSHIP / n+1 카드게임

Author: himitery

programmers/2024_KAKAO_WINTER_INTERNSHIP/python/n_plus_1_카드게임/__init__.py

@@ -0,0 +1,33 @@
+from typing import List, Set
+
+
+def solution(coin: int, cards: List[int]) -> int:
+    n, count = len(cards), 1
+    init, picked, remained = set(cards[: n // 3]), set(), cards[n // 3 :][::-1]
+
+    while remained:
+        picked |= {remained.pop(), remained.pop()}
+
+        found = False
+        for x, y, amount in [(init, init, 0), (init, picked, 1), (picked, picked, 2)]:
+            if amount > coin:
+                break
+
+            if (pair := find_pair(n + 1, x, y)) and pair:
+                found = True
+                init, picked = init - pair, picked - pair
+                count, coin = count + 1, coin - amount
+                break
+
+        if not found:
+            break
+
+    return count
+
+
+def find_pair(target: int, x: Set[int], y: Set[int]) -> Set[int]:
+    for card in x:
+        if target - card in y:
+            return {card, target - card}
+
+    return set()

programmers/2024_KAKAO_WINTER_INTERNSHIP/python/n_plus_1_카드게임/main.py

@@ -0,0 +1,88 @@
+[
+  {
+    "params": [
+      4,
+      [
+        3,
+        6,
+        7,
+        2,
+        1,
+        10,
+        5,
+        9,
+        8,
+        12,
+        11,
+        4
+      ]
+    ],
+    "expected": 5
+  },
+  {
+    "params": [
+      3,
+      [
+        1,
+        2,
+        3,
+        4,
+        5,
+        8,
+        6,
+        7,
+        9,
+        10,
+        11,
+        12
+      ]
+    ],
+    "expected": 2
+  },
+  {
+    "params": [
+      2,
+      [
+        5,
+        8,
+        1,
+        2,
+        9,
+        4,
+        12,
+        11,
+        3,
+        10,
+        6,
+        7
+      ]
+    ],
+    "expected": 4
+  },
+  {
+    "params": [
+      10,
+      [
+        1,
+        2,
+        3,
+        4,
+        5,
+        6,
+        7,
+        8,
+        9,
+        10,
+        11,
+        12,
+        13,
+        14,
+        15,
+        16,
+        17,
+        18
+      ]
+    ],
+    "expected": 1
+  }
+]

programmers/2024_KAKAO_WINTER_INTERNSHIP/python/n_plus_1_카드게임/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()