[programmers/py] [PCCE 기출문제] 9번 / 이웃한 칸

승민·2024년 3월 15일

알고리즘

목록 보기
78/171

[PCCE 기출문제] 9번 / 이웃한 칸

https://school.programmers.co.kr/learn/courses/30/lessons/250125

문제 설명

  • 보드의 각 칸에 칠해진 색깔 이름이 담긴 이차원 문자열 리스트 board와 고른 칸의 위치를 나타내는 두 정수 h, w가 주어질 때 board[h][w]와 이웃한 칸들 중 같은 색으로 칠해져 있는 칸의 개수를 return 하도록 solution 함수를 완성해 주세요.

문제 풀이

def solution(board, h, w):
    # 위, 아래, 왼, 오
    x = [0, 0, -1, 1]
    y = [-1, 1, 0, 0]
    
    answer = 0
    now = board[h][w] # color
    
    height = len(board)
    width = len(board[0])
    
    for i in range(4):
        nx = h + x[i]
        ny = w + y[i]
        
        # 정상 적인 범위
        if nx >= 0 and ny >=0 and nx < width and ny < height :
            if board[nx][ny] == now:
                answer += 1
    
    return answer

0개의 댓글