solved(python): programmers

- 2023 KAKAO BLIND RECRUITMENT / 표현 가능한 이진트리

Author: himitery

programmers/2023_KAKAO_BLIND_RECRUITMENT/python/표현_가능한_이진트리/__init__.py

@@ -0,0 +1,40 @@
+import math
+from collections import deque
+from typing import List
+
+
+def solution(numbers: List[int]) -> List[int]:
+    return [int(is_binary(num)) for num in numbers]
+
+
+def is_binary(num: int) -> bool:
+    binary = to_binary(num)
+    queue, count = deque([(1, len(binary))]), 0
+
+    while queue:
+        left, right = queue.popleft()
+        mid = (left + right) // 2
+
+        if binary[mid - 1] == 0:
+            continue
+
+        count += 1
+        if left < mid:
+            queue.append((left, mid - 1))
+        if mid < right:
+            queue.append((mid + 1, right))
+
+    return count == binary.count(1)
+
+
+def to_binary(num: int) -> List[int]:
+    binary = []
+    while num:
+        binary.append(num & 1)
+        num >>= 1
+
+    size = len(binary)
+    for _ in range(size + 1, 2 ** (int(math.log2(size)) + 1)):
+        binary.append(0)
+
+    return binary[::-1]

programmers/2023_KAKAO_BLIND_RECRUITMENT/python/표현_가능한_이진트리/main.py

@@ -0,0 +1,30 @@
+[
+  {
+    "params": [
+      [
+        7,
+        42,
+        5
+      ]
+    ],
+    "expected": [
+      1,
+      1,
+      0
+    ]
+  },
+  {
+    "params": [
+      [
+        63,
+        111,
+        95
+      ]
+    ],
+    "expected": [
+      1,
+      1,
+      0
+    ]
+  }
+]

programmers/2023_KAKAO_BLIND_RECRUITMENT/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()