[Python] 백준 7562번 - 나이트의 이동 (실버1)

https://www.acmicpc.net/problem/7562

7562번: 나이트의 이동

 

 

혼자 힘으로 풀었는가? O

알고리즘 분류
 - 그래프 이론
 - 그래프 탐색
 - 너비 우선 탐색

 

문제

체스판 위에 한 나이트가 놓여져 있다. 나이트가 한 번에 이동할 수 있는 칸은 아래 그림에 나와있다. 나이트가 이동하려고 하는 칸이 주어진다. 나이트는 몇 번 움직이면 이 칸으로 이동할 수 있을까?

입력

입력의 첫째 줄에는 테스트 케이스의 개수가 주어진다.

각 테스트 케이스는 세 줄로 이루어져 있다. 첫째 줄에는 체스판의 한 변의 길이 l(4 ≤ l ≤ 300)이 주어진다. 체스판의 크기는 l × l이다. 체스판의 각 칸은 두 수의 쌍 {0, ..., l-1} × {0, ..., l-1}로 나타낼 수 있다. 둘째 줄과 셋째 줄에는 나이트가 현재 있는 칸, 나이트가 이동하려고 하는 칸이 주어진다.

출력

각 테스트 케이스마다 나이트가 최소 몇 번만에 이동할 수 있는지 출력한다.

 

---

 나이트의 이동 동선은 아래와 같다.

 

 

 

이는

dir = [(-2, -1), (-2, 1), (-1, 2), (1, 2), (2, 1), (2, -1), (1, -2), (-1, -2)]

위와 같이 정리 가능하다.

 

그 다음 각 지점을 방문하는 모든 구간을 기존 값+1을 해주며 목적지가 나올 때까지 반복한다.

 

import sys
input = sys.stdin.readline
 
from collections import deque
 
T = int(input())
 
dir = [(-2, -1), (-2, 1), (-1, 2), (1, 2), (2, 1), (2, -1), (1, -2), (-1, -2)]
 
def bfs(start, end, data):
    q = deque()
    q.append((start[0], start[1]))
    data[start[0]][start[1]] = 1
    while q:
        now_x, now_y = q.popleft()
        cnt = data[now_x][now_y]
        for i in range(8):
            dx, dy = dir[i]
            nx = now_x + dx
            ny = now_y + dy
            if 0 <= nx < n and 0 <= ny < n:
                if nx == end[0] and ny == end[1]:
                    print(cnt)
                    return
                if data[nx][ny] == 0:
                    data[nx][ny] = cnt+1
                    q.append((nx, ny))
 
for _ in range(T):
    n = int(input())
    data = [ [0] * n for _ in range(n)]
    start = list(map(int, input().split()))
    end = list(map(int, input().split()))
    
    if start == end:
        print(0)
    else:
        bfs(start, end, data)