프로그래머스: 이웃한 칸

[level unrated] [PCCE 기출문제] 9번 / 이웃한 칸 - 250125

문제 링크

성능 요약

메모리: 10.4 MB, 시간: 0.01 ms

구분

코딩테스트 연습 > PCCE 기출문제

채점결과

정확성: 100.0
합계: 100.0 / 100.0

제출 일자

2024년 1월 4일 13:39:18

문제 설명

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

보드의 각 칸에 칠해진 색깔 이름이 담긴 이차원 문자열 리스트 board와 고른 칸의 위치를 나타내는 두 정수 h, w가 주어질 때 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개입니다.

---

• cpp를 응시하는 경우 리스트는 배열과 동일한 의미이니 풀이에 참고해주세요.
  • ex) 번호가 담긴 정수 리스트 numbers가 주어집니다. => 번호가 담긴 정수 배열 numbers가 주어집니다.
• java를 응시하는 경우 리스트는 배열, 함수는 메소드와 동일한 의미이니 풀이에 참고해주세요.
  • ex) solution 함수가 올바르게 작동하도록 한 줄을 수정해 주세요. => solution 메소드가 올바르게 작동하도록 한 줄을 수정해 주세요.

풀이

def solution(board, h, w):
    answer = 0
    dh, dw = [0 ,1 ,-1, 0], [1, 0, 0, -1]
    n = len(board)
    looking_for = board[h][w]

    for i in range(4):
        h_check, w_check = h + dh[i], w + dw[i]

        if in_range(h_check, w_check, n):
            if board[h_check][w_check] == looking_for:
                answer += 1

    return answer

def in_range(dh, dw, n):
    if (dh >= 0 and dh < n) and (dw >= 0 and dw < n):
        return True
    return False

느낀점

예전에 알고리즘 문제풀이에서 dx, dy 이동에 대한 테크닉을 배운게 있어서 다시 한번 짧게 복습 후 해당 문제를 풀이했다. 주어진 보드에 h, w에 위치한 곳의 색을 찾아내고, 좌우상하로 움직이면서 해당 칸이 in_range 범위 안에 존재하고 색이 같다면 answer에 1씩 더해주면 되는 문제이다.

격자 문제에서는 항상 내가 움직일 수 있는지를 확인하는게 제일 포인트인것 같다.