코테준비/프로그래머스

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

예찬예찬 2023. 11. 24. 22:55
728x90
반응형

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

 

프로그래머스

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

programmers.co.kr

 

아이디어

  • DFS에서 사용한 로직을 그대로 쓰면 될 것 같다.
  • 4방향을 가리킬 배열 dy, dx를 만들어 반복문을 돌려 탐색
  • 범위를 벗어나는 좌표인지 판단 후  벗어나지 않으면 원본과 비슷한지 확인한다.
  • 원본과 같으면 answer + 1

코드

def solution(board, h, w):
    dy = [0,-1,0,1]
    dx = [-1,0,1,0]
    answer = 0
    
    for i in range(4):
        nh = h + dy[i]
        nw = w + dx[i]
        if 0<= nh < len(board) and 0<= nw < len(board[0]):
            if board[nh][nw] == board[h][w]:
                answer += 1
    
    return answer

 

마무리

  • 오랜만에 푸는 알고리즘인데 기본적인 것은 기억하고 있는 것 같다.
  • 레벨 1 문제는 양심이 없었다 다음부터는 최소 2 레벨 문제 풀게요...

 

728x90
반응형