2023 KAKAO BLIND RECRUITMENT 표 병합

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/150366

Author: skysign

2023 KAKAO BLIND RECRUITMENT 표 병합/main.py

@@ -0,0 +1,62 @@
+from typing import List
+
+
+def solution(commands: List[str]):
+    MX = 51
+    map = [['EMPTY' for _ in range(MX)] for _ in range(MX)]
+    merged = [[(r, c) for c in range(MX)] for r in range(MX)]
+    answer = []
+
+    for cmd in commands:
+        cmd = cmd.split(' ')
+
+        if cmd[0] == 'UPDATE' and len(cmd) == 4:
+            r, c, v = int(cmd[1]), int(cmd[2]), cmd[3]
+            if merged[r][c] == (r,c):
+                map[r][c] = v
+            else:
+                mr, mc = merged[r][c]
+                map[mr][mc] = v
+
+        elif cmd[0] == 'UPDATE' and len(cmd) == 3:
+            _, v1, v2 = cmd
+
+            for r in range(1, MX):
+                for c in range(1, MX):
+                    if map[r][c] == v1:
+                        map[r][c] = v2
+
+        elif cmd[0] == 'MERGE':
+            r1, c1, r2, c2 = int(cmd[1]), int(cmd[2]), int(cmd[3]), int(cmd[4])
+            to = merged[r1][c1]
+            fr = merged[r2][c2]
+            if map[to[0]][to[1]] == 'EMPTY':
+                map[to[0]][to[1]] = map[fr[0]][fr[1]]
+
+            for r in range(1, MX):
+                for c in range(1, MX):
+                    if merged[r][c] == fr:
+                        merged[r][c] = to
+
+        elif cmd[0] == 'UNMERGE':
+            sr, sc = int(cmd[1]), int(cmd[2])
+            mr, mc = merged[sr][sc]
+            v = map[mr][mc]
+
+            for r in range(1, MX):
+                for c in range(1, MX):
+                    if merged[r][c] == (mr, mc):
+                        merged[r][c] = (r, c)
+                        map[r][c] = 'EMPTY'
+
+            map[sr][sc] = v
+
+        elif cmd[0] == 'PRINT':
+            r, c = int(cmd[1]), int(cmd[2])
+            if merged[r][c] != (r, c):
+                mr, mc = merged[r][c]
+                answer.append(map[mr][mc])
+            else:
+                answer.append(map[r][c])
+
+    return answer

2023 KAKAO BLIND RECRUITMENT 표 병합/test_main.py

@@ -0,0 +1,19 @@
+from unittest import TestCase
+from main import solution
+
+
+class Test(TestCase):
+    def test1_solution(self):
+        self.assertEqual(["EMPTY", "group"], solution(
+            ["UPDATE 1 1 menu", "UPDATE 1 2 category", "UPDATE 2 1 bibimbap", "UPDATE 2 2 korean", "UPDATE 2 3 rice", "UPDATE 3 1 ramyeon", "UPDATE 3 2 korean", "UPDATE 3 3 noodle",
+             "UPDATE 3 4 instant", "UPDATE 4 1 pasta", "UPDATE 4 2 italian", "UPDATE 4 3 noodle", "MERGE 1 2 1 3", "MERGE 1 3 1 4", "UPDATE korean hansik", "UPDATE 1 3 group", "UNMERGE 1 4",
+             "PRINT 1 3", "PRINT 1 4"]))
+
+    def test2_solution(self):
+        self.assertEqual(["d", "EMPTY"],
+                         solution(["UPDATE 1 1 a", "UPDATE 1 2 b", "UPDATE 2 1 c", "UPDATE 2 2 d", "MERGE 1 1 1 2", "MERGE 2 2 2 1", "MERGE 2 1 1 1", "PRINT 1 1", "UNMERGE 2 2", "PRINT 1 1"]))
+
+    # 출처: https://school.programmers.co.kr/questions/71705
+    def testx_solution(self):
+        self.assertEqual(["EMPTY"],
+                         solution(["MERGE 1 1 2 2", "PRINT 1 1"]))