[LV1] 이웃한 칸 - PCCE 기출문제

* 저 만의 문제풀이라 깔끔하지 않을 수도 있으니 참고부탁드립니다.

Q. 각 칸마다 색이 칠해진 2차원 격자 보드판에서 좌표값 h, w가 주어질 때 이웃한 칸들 중 같은 색으로 칠해져 있는 칸의 개수는?

• 이웃한 칸은 위, 아래, 양 옆

def solution(board, h, w):
    bn=len(board)-1
    clr=board[h][w]
    clr_idx=0
    if h>0:
        if board[h-1][w]==clr:
            clr_idx+=1
    if h<bn:
        if board[h+1][w]==clr:
            clr_idx+=1
    if w>0:
        if board[h][w-1]==clr:
            clr_idx+=1
    if w<bn:
        if board[h][w+1]==clr:
            clr_idx+=1
    return clr_idx