python | 나이트의 이동[백준 7562]

나경호·2022년 4월 9일

BFS boj python

알고리즘 Algorithm

목록 보기

55/106

나이트의 이동

출처 | 나이트의 이동[백준 7562]

문제

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

입력

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

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

출력

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

풀이

from sys import stdin
from collections import deque

def bfs(n, startY, startX, endY, endX, visited):
    dr = ((2, 1), (1, 2), (-2, 1), (-1, 2), (-2, -1), (-1, -2), (2, -1), (1, -2))
    dq = deque()

    dq.append((startY, startX))

    if startX == endX and startY == endY:
        return 0
    
    while dq:
        y, x = dq.popleft()

        for dy, dx in dr:
            ny = y + dy
            nx = x + dx

            if ny < 0 or nx < 0 or ny >= n or nx >= n:
                continue
            
            if visited[ny][nx] == 0:
                visited[ny][nx] = visited[y][x] + 1
                dq.append((ny, nx))

            if ny == endY and nx == endX:
                return visited[ny][nx]

t = int(input())
for _ in range(t):
    w = int(input())
    startY, startX = map(int, stdin.readline().split())
    endY, endX = map(int, stdin.readline().split())
    visited = [[0] * w for _ in range(w)]
    print(bfs(w, startY, startX, endY, endX, visited))