코딩 테스트/Level 1

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

컴닥 2024. 4. 3. 14:09
반응형

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

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

print(solution(
    [["blue", "red", "orange", "red"],
     ["red", "red", "blue", "orange"],
     ["blue", "orange", "red", "red"],
     ["orange", "orange", "red", "blue"]],
    0,
    0
))
반응형