[프로그래머스] [PCCE 기출문제] 9번 / 이웃한 칸

Chobby·2024년 2월 25일
1

Programmers

목록 보기
336/349

def solution(board, h, w):
    n = len(board)
    count = 0
    targetColor = board[h][w]
    # 우 하 상 좌 방향
    dh = [0, 1, -1, 0]
    dw = [1, 0, 0, -1]
    for i in range(0, 4):
        h_check = h + dh[i]
        w_check = w + dw[i]
        # 범위검사         
        if((h_check < 0 or h_check >= n)):
            continue
        if(w_check < 0 or w_check >= n):
            continue
        # 색상비교
        if(targetColor == board[h_check][w_check]):
            count += 1
    return count
profile
내 지식을 공유할 수 있는 대담함

0개의 댓글