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

2024. 9. 9. 18:00PCCE 기출문제

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

 

프로그래머스

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

programmers.co.kr


문제 설명
각 칸마다 색이 칠해진 2차원 격자 보드판이 있습니다. 그중 한 칸을 골랐을 때, 위, 아래, 왼쪽, 오른쪽 칸 중 같은 색깔로 칠해진 칸의 개수를 구하려고 합니다.

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

이웃한 칸들 중 몇 개의 칸이 같은 색으로 색칠되어 있는지 확인하는 과정은 다음과 같습니다.

1. 정수를 저장할 변수 n을 만들고 board의 길이를 저장합니다.

2. 같은 색으로 색칠된 칸의 개수를 저장할 변수 count를 만들고 0을 저장합니다.

3. h와 w의 변화량을 저장할 정수 리스트 dh, dw를 만들고 각각 [0, 1, -1, 0], [1, 0, 0, -1]을 저장합니다.

4. 반복문을 이용해 i 값을 0부터 3까지 1 씩 증가시키며 아래 작업을 반복합니다.

    4-1. 체크할 칸의 h, w 좌표를 나타내는 변수 h_check, w_check를 만들고 각각 h + dh[i], w + dw[i]를 저장합니다.

    4-2. h_check가 0 이상 n 미만이고 w_check가 0 이상 n 미만이라면 다음을 수행합니다.
        4-2-a. board[h][w]와 board[h_check][w_check]의 값이 동일하다면 count의 값을 1 증가시킵니다.

5. count의 값을 return합니다.
  • 위의 의사코드와 작동방식이 다른 코드를 작성해도 상관없습니다.

제한사항

  • 1 ≤ board의 길이 ≤ 7
    • board의 길이와 board[n]의 길이는 동일합니다.
  • 0 ≤ h, w < board의 길이
  • 1 ≤ board[h][w]의 길이 ≤ 10
    • board[h][w]는 영어 소문자로만 이루어져 있습니다.

입출력 예

board h w result
[
     ["blue", "red", "orange", "red"], 
     ["red", "red", "blue", "orange"], 
     ["blue", "orange", "red", "red"], 
     ["orange", "orange", "red", "blue"]
]
1 1 2
[
     ["yellow", "green", "blue"],
     ["blue", "green", "yellow"],
     ["yellow", "blue", "blue"]
]
0 1 1

입출력 예 설명
입출력 예 #1

  • 주어진 보드를 나타내면 아래 그림과 같고, board[1][1]의 인접한 칸들은 화살표가 가리키는 칸들입니다. 따라서 board[1][1] 주변의 "red" 칸은 총 2개입니다.

입출력 예 #2

  • 주어진 보드를 나타내면 아래 그림과 같고, board[0][1]의 인접한 칸들은 화살표가 가리키는 칸들입니다. 따라서 board[0][1] 주변의 "green"칸은 총 1개입니다.


특정 위치 board[h][w]의 인접 위치에 있는 보드들이 같은 색을 가지고 있는지 구하는 문제입니다.

 

문제 내에서 어떻게 풀어야 하는지 잘 설명해주었기 때문에 딱히 추가 설명이 필요하지는 않을 것 같네요.

이중 리스트로 구현해도 되는데, 이건 개인 취향이니 넘어가겠습니다.

 

저는 이중 리스트를 사용해서 문제를 풀어보겠습니다.

def solution(board, h, w):
    # 개수를 셀 변수 answer
    answer = 0
    # 방향을 나타내는 directions
    directions = [[0, 1], [1, 0], [0, -1], [-1, 0]]
    # 모든 방향에 대해서 탐색을 진행합니다.
    for dx, dy in directions:
        # 다음 위치 설정
        tx, ty = h + dx, w + dy
        # 해당 위치가 정상적인 위치인지 확인합니다.
        # 정상 위치일 때 색이 같은지 확인합니다.
        if 0 <= tx < len(board) and 0 <= ty < len(board) and board[tx][ty] == board[h][w]:
            # 색이 같다면 answer에 1을 추가합니다.
            answer += 1
    return answer

 

1. 탐색을 위해 directions을 선언합니다. 저는 이중 리스트로 방향을 설정했습니다.

2. 반복문을 사용해서 모든 방향을 탐색합니다.

3. 방향으로 진행했을 때 정상 범위인지 확인합니다.

4. 정상 범위라면 해당 위치의 색이 board[h][w]와 같은 색인지 확인합니다.

5. 같다면 1을 추가하고, 아니라면 다음 방향으로 넘어갑니다.

 

어떻게 보면 이 문제는 BFS 알고리즘에서 추가 탐색 (큐 입력) 을 제외하고 현재 범위만 탐색하는 문제라고도 볼 수 있겠네요.

어려운 문제는 아니었습니다.