빠른 오름차순 숫자 탐색

난이도

Gold 5

인증

문제

5 x 5 크기의 보드가 주어진다. 보드는 1 x 1 크기의 정사각형 격자로 이루어져 있다. 보드의 격자에는 -1, 0, 1, 2, 3, 4, 5, 6중 하나의 수가 적혀 있다. 격자의 위치는 (r, c)로 표시한다. r은 행 번호, c는 열 번호를 나타낸다. 행 번호는 맨 위 위치가 0이고 아래 방향으로 1씩 증가한다. 열 번호는 맨 왼쪽 위치가 0이고 오른쪽으로 1씩 증가한다. 즉, 맨 왼쪽 위 위치가 (0, 0), 맨 아래 오른쪽 위치가 (4, 4)이다. -1이 적혀 있는 칸으로는 이동할 수 없고 0, 1, 2, 3, 4, 5, 6이 적혀 있는 칸으로는 이동할 수 있다.

현재 한 명의 학생이 (r, c) 위치에 있고 한 번의 이동으로 상, 하, 좌, 우 방향 중에서 한가지 방향으로 한 칸 이동할 수 있다. 학생이 현재 위치 (r, c)에서 시작하여 1, 2, 3, 4, 5, 6이 적혀 있는 칸을 순서대로 방문하려고 한다. 보드에는 1, 2, 3, 4, 5, 6이 적혀 있는 칸이 1개씩 존재하고 1, 2, 3, 4, 5, 6이 적혀 있는 칸을 여러 번 방문할 수 있다. 즉, 1이 적혀 있는 칸에서 2가 적혀 있는 칸으로 이동하고, 2가 적혀 있는 칸에서 3이 적혀 있는 칸으로 이동하고, … , 5가 적혀 있는 칸에서 6이 적혀 있는 칸으로 이동한다. i가 적혀 있는 칸에서 i + 1이 적혀 있는 칸으로 이동할 때 다른 번호가 적힌 칸을 방문해도 된다.(1 ≤ i ≤ 5) 마찬가지로, 현재 위치 (r, c)에서 1이 적혀 있는 칸으로 이동할 때 다른 번호가 적힌 칸을 방문해도 된다. 학생이 현재 위치 (r, c)에서 시작하여 1, 2, 3, 4, 5, 6이 적혀 있는 칸을 순서대로 방문하는 최소 이동 횟수를 출력하자. 학생이 현재 위치 (r, c)에서 시작하여 1, 2, 3, 4, 5, 6이 적혀 있는 칸을 순서대로 방문할 수 없는 경우 -1을 출력한다.

입력

첫 번째 줄부터 다섯 개의 줄에 걸쳐 보드의 각 칸에 적혀있는 수가 순서대로 주어진다. i번째 줄의 j번째 수는 보드의 (i - 1)번째 행, (j - 1)번째 열에 적혀있는 수를 나타낸다. 보드의 각 칸에 적혀 있는 수는 -1, 0, 1, 2, 3, 4, 5, 6중 하나이다.

다음 줄에 학생의 현재 위치 r, c가 빈칸을 사이에 두고 순서대로 주어진다.

출력

학생이 현재 위치 (r, c)에서 시작하여 1, 2, 3, 4, 5, 6이 적혀 있는 칸을 순서대로 방문하는 최소 이동 횟수를 출력한다. 학생이 현재 위치 (r, c)에서 시작하여 1, 2, 3, 4, 5, 6이 적혀 있는 칸을 순서대로 방문할 수 없는 경우 -1을 출력한다.

예제 입력

0 0 1 0 0
0 0 2 0 0
0 0 3 0 0
0 0 4 0 0
0 0 5 6 -1
0 1

예제 출력

6

해설 및 후기

빠른 오름차순 메시지 전달과 비슷한 문제같아 보이지만 실상은 많이 다른 문제이다. 상하좌우로 이동하면서 최소 이동 경로를 찾는 문제이다. 떄문에 자연스럽게 bfs를 선택하였으나, 이전 경로를 다시 방문할 수 있음에 유의하여야 했다. 때문에 목표로 하는 번호 정보를 추가하여, 어떤 번호를 찾고 있을때 방문했는지를 판별하도록 하여 이를 해결했다.

제출 코드

import sys
from collections import deque
isVisit = [[[False for _ in range(6)] for _ in range(5)] for _ in range(5)]
q = deque()
m = []
for i in range(5):
    m.append(list(map(int, sys.stdin.readline().rstrip().split())))
ix, iy = map(int, input().split())
q.appendleft((ix, iy, 0, 1)) #마지막은 Tofind
cal = [[1,0],[-1,0],[0,1],[0,-1]]
while(q):
    x,y,val,toFind = q.pop()
    chk = 0
    if(isVisit[x][y][toFind-1]): continue
    isVisit[x][y][toFind-1] = True
    if(m[x][y] == -1): continue
    if(toFind == 6 and m[x][y] == 6):
        print(val)
        exit()
    if(m[x][y] == toFind):
        chk += 1
    for dx,dy in cal:
        dxx = dx+x
        dyy = dy+y
        if(dxx >= 0 and dxx < 5 and dyy >= 0 and dyy < 5):
            q.appendleft((dxx,dyy,val+1,toFind+chk))
print(-1)