관리 메뉴

커리까지

[백준] 7562번 나이트의 이동 파이썬 본문

알고리즘/풀이

[백준] 7562번 나이트의 이동 파이썬

목표는 커리 2023. 7. 22. 17:02
728x90
SMALL

문제

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

img

입력

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

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

출력

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

예제 입력 1
3
8
0 0
7 0
100
0 0
30 50
10
1 1
1 1
예제 출력 1
5
28
0

제출 답안

from collections import deque
import sys
input = sys.stdin.readline

dy = [1,2,2,1,-1,-2,-2,-1]
dx = [2,1,-1,-2,-2,-1,1,2]

t = int(input())
answer = []

def bfs(graph,a, b, j, i):
    n = len(graph)
    q = deque([(a, b)])
    while q:
        y, x = q.popleft()
        if y == j and x == i:
            return graph[j][i]
        for k in range(8):
            ny = y + dy[k]
            nx = x + dx[k]
            if 0 <= ny < n and 0 <= nx < n:
                if graph[ny][nx] == 0:
                    graph[ny][nx] = graph[y][x] + 1
                    q.append((ny, nx))
    return graph[j][i]

for _ in range(t):
    n = int(input())
    graph = [[0] * n for _ in range(n)]
    a, b = map(int, input().split())
    graph[a][b] = 1
    y, x = map(int, input().split())
    cnt = bfs(graph, a, b, y ,x)
    answer.append(cnt-1)
print(*answer, sep='\n')
728x90
LIST
Comments