Python: 24 PCCE 기출문제 9번

Author: AMinSC

data/py/lv1/Lv1_2024_PCCE_기출문제_9번_이웃한_칸_구현_성철.py

@@ -0,0 +1,13 @@
+def solution(board, h, w):
+    answer = 0
+    color = board[h][w]
+    n = len(board)
+    dh, dw = [0, 1, -1, 0], [1, 0, 0, -1]
+    
+    for i in range(4):
+        h_check, w_check = h + dh[i], w + dw[i]
+        if 0 <= h_check < n and 0 <= w_check < n:
+            if color == board[h_check][w_check]:
+                answer += 1
+    
+    return answer