[프로그래머스/PCCP_9번] 이웃한 칸 ( 파이썬 )

이번 문제는 시작 지점에서 주위에 색이 있는 칸의 수를 구하는 문제입니다.

입력과 출력 예시

입출력 예 설명

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