IT/코딩테스트

[IT/코딩테스트] KB ITYL 중간평가 대비 : 보물지도 (programmers)

토끼개발자 조르디 2025. 5. 29. 17:48

 

▶ 문제설명

진수는 보물이 묻힌 장소와 함정이 표시된 보물 지도를 이용해 보물을 찾으려 합니다. 보물지도는 가로 길이가 n, 세로 길이가 m인 직사각형 모양입니다.

맨 왼쪽 아래 칸의 좌표를 (1, 1)으로, 맨 오른쪽 위 칸의 좌표를 (n, m)으로 나타냈을 때, 보물은 (n, m) 좌표에 묻혀있습니다. 또한, 일부 칸에는 함정이 있으며, 해당 칸으로는 지나갈 수 없습니다.

진수는 처음에 (1, 1) 좌표에서 출발해 보물이 있는 칸으로 이동하려 합니다. 이동할 때는 [상,하,좌,우]로 인접한 네 칸 중 한 칸으로 걸어서 이동합니다. 한 칸을 걸어서 이동하는 데 걸리는 시간은 1입니다.

진수는 보물이 위치한 칸으로 수월하게 이동하기 위해 신비로운 신발을 하나 준비했습니다. 이 신발을 신고 뛰면 한 번에 두 칸을 이동할 수 있으며, 함정이 있는 칸도 넘을 수 있습니다. 하지만, 이 신발은 한 번밖에 사용할 수 없습니다. 신비로운 신발을 사용하여 뛰어서 두 칸을 이동하는 시간도 1입니다.

이때 진수가 출발점에서 보물이 위치한 칸으로 이동하는데 필요한 최소 시간을 구해야 합니다.

예를 들어, 진수의 보물지도가 아래 그림과 같을 때, 진수가 걸어서 오른쪽으로 3칸, 걸어서 위쪽으로 3칸 이동하면 6의 시간에 보물이 위치한 칸으로 이동할 수 있습니다. 만약, 오른쪽으로 걸어서 1칸, 위쪽으로 걸어서 1칸, 신비로운 신발을 사용하여 위로 뛰어 2칸, 오른쪽으로 걸어서 2칸 이동한다면 총 5의 시간에 보물이 위치한 칸으로 이동할 수 있으며, 이보다 빠른 시간 내에 보물이 있는 위치에 도착할 수 없습니다.

 

 

진수의 보물지도가 표현하는 지역의 가로 길이를 나타내는 정수 n, 세로 길이를 나타내는 정수 m, 함정의 위치를 나타내는 2차원 정수 배열 hole이 주어졌을 때, 진수가 보물이 있는 칸으로 이동하는데 필요한 최소 시간을 return 하는 solution 함수를 완성해주세요. 단, 보물이 있는 칸으로 이동할 수 없다면, -1을 return 해야 합니다.

 

 

▶ 제한사항

⦁ 1 ≤ n, m ≤ 1,000
     ∘ 단, n * m이 3 이상인 경우만 입력으로 주어집니다.

⦁ 1 ≤ hole의 길이 ≤ n * m - 2
     ∘ hole[i]는 [a, b]의 형태이며, (a, b) 칸에 함정이 존재한다는 의미이며, 1 ≤ a ≤ n, 1 ≤ b ≤ m을 만족합니다.
     ∘ 같은 함정에 대한 정보가 중복해서 들어있지 않습니다.

⦁ (1, 1) 칸과 (n, m) 칸은 항상 함정이 없습니다.


▶ 입출력 예

n m hole result
4 4 [[2, 3], [3, 3]] 5
5 4 [[1, 4], [2, 1], [2, 2], [2, 3], [2, 4], [3, 3], [4, 1], [4, 3], [5, 3]] -1


▶ 입출력 예 설명

입출력 예 #1
본문의 예시와 같습니다.

 


입출력 예 #2
보물지도를 그림으로 나타내면 아래와 같으며, 보물이 위치한 칸으로 이동할 수 없습니다. 따라서, -1을 return 해야 합니다.

 

문제 정리

  • 문제 유형: BFS ( 너비 우선 탐색 )
  • 풀이 로직 요약:
    1. 각각의 칸에서의 방문 여부를 jump를 썼는가 안 썼는가 상태까지 포함하여 관리
      • visited[row][col][0]: 점프 안 한 상태에서 방문
      • visited[row][col][1]: 점프 한 상태에서 방문
    2. 일반 이동(1칸)과 점프 이동(2칸)을 구분하여 큐에 삽입
    3. 점프는 한 번만 허용되기 때문에, 점프 여부도 상태에 포함
    4. 목적지 (n, m)에 도달하면 현재까지의 거리 반환
  • 깨달은 점 / 핵심:
    • 일반적인 2차원 visited 배열이 아니라, [n+1][m+1][2]의 3차원 visited를 사용해야 문제를 올바르게 풀기 가능
    • 복잡한 BFS 문제를 해결할 때는 모든 가능한 상태를 방문체크하고 큐에 넣는 것이 핵심
    • 점프처럼 조건이 있는 기능은 그 기능을 사용했는지 여부까지 상태로 모델링 필요

 

솔루션 코드

from collections import deque

def solution(n, m, hole):
    trap = [[False] * (m + 1) for _ in range(n + 1)]
    for r, c in hole:
        trap[r][c] = True

    visited = [[[False] * 2 for _ in range(m + 1)] for _ in range(n + 1)]
    queue = deque()
    queue.append((1, 1, 0, 0))  # (r, c, jumped(0 or 1), distance)
    visited[1][1][0] = True

    # 4방향
    dir1 = [(-1, 0), (1, 0), (0, -1), (0, 1)]   # 1칸
    dir2 = [(-2, 0), (2, 0), (0, -2), (0, 2)]   # 점프(2칸)

    while queue:
        r, c, jumped, dist = queue.popleft()

        # 목적지 도달 시
        if r == n and c == m:
            return dist

        # 일반 1칸 이동
        for dr, dc in dir1:
            nr, nc = r + dr, c + dc
            if 1 <= nr <= n and 1 <= nc <= m and not visited[nr][nc][jumped] and not trap[nr][nc]:
                visited[nr][nc][jumped] = True
                queue.append((nr, nc, jumped, dist + 1))

        # 점프 이동 (아직 점프 안 했을 때만 가능)
        if jumped == 0:
            for dr, dc in dir2:
                nr, nc = r + dr, c + dc
                if 1 <= nr <= n and 1 <= nc <= m and not visited[nr][nc][1] and not trap[nr][nc]:
                    visited[nr][nc][1] = True
                    queue.append((nr, nc, 1, dist + 1))

    return -1  # 목적지 도달 불가