코딩 테스트

백준 : 나이트의 이동 [파이썬, bfs, silver1]

Squidward 2023. 3. 11. 21:51

문제

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

입력

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

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

출력

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

예제 입력 

3
8
0 0
7 0
100
0 0
30 50
10
1 1
1 1

예제 출력 

5
28
0

 

[문제 풀이]

다른 bfs 유형 문제와 비슷하다.

이동 거리를 지정해준다는 것이 특징인데, 좌표로 만들어서 반복문 돌리자

# 한번에 이동 가능한 좌표
dxy = [(1,2),(2,1),(-1,-2),(-2,-1),(-1,2),(1,-2),(-2,1),(2,-1)]

 

[정답 코드]

import sys
from collections import deque

def bfs(nx, ny, mx, my):
    queue = deque()
    queue.append((nx,ny))
    visited[nx][ny] = 1

    while queue:
        a,b = queue.popleft()

        if a==mx and b==my:
            print(visited[a][b]-1)

        # 이동 좌표 모두 적용
        for i,j in dxy:
            tx = a + i
            ty = b + j

            # 범위 내에 있고, 방문하기 전이라면
            if (0<=tx<num) and (0<=ty<num) and not visited[tx][ty]:
                queue.append((tx,ty))
                visited[tx][ty] = visited[a][b] + 1

# 한번에 이동 가능한 좌표
dxy = [(1,2),(2,1),(-1,-2),(-2,-1),(-1,2),(1,-2),(-2,1),(2,-1)]
# 테스트 케이스 수
n = int(sys.stdin.readline())

for i in range(n):
    num = int(sys.stdin.readline())
    visited = [[0 for _ in range(num)] for _ in range(num)]
    nx, ny = map(int, input().split()) # 현재 위치
    mx, my = map(int, input().split()) # 목표 위치
    bfs(nx, ny, mx, my)
728x90