Algorithm PS/DFS & BFS

[BOJ/백준] 7562번 나이트의 이동 (Python 파이썬)

great_park 2023. 1. 27. 17:23
반응형

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

 

7562번: 나이트의 이동

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

www.acmicpc.net

 

문제

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

입력

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

각 테스트 케이스는 세 줄로 이루어져 있다. 첫째 줄에는 체스판의 한 변의 길이 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

 


풀이

나이트의 이동 가능한 수는 위의 그림과 같이 총 8개입니다. 이를 x, y의 변화량으로 표현하면 아래와 같습니다.

# 나이트의 이동 경우의 수
dx = [1, 2, 2, 1, -1, -2, -2, -1]
dy = [2, 1, -1, -2, -2, -1, 1, 2]

이를 바탕으로 BFS내에서 주어진 점에서 총 8개의 수를 고려하는 식으로 진행하면 됩니다. 종료 조건은, input으로 주어진목표 지점에 도착하면 탐색을 종료하고, 방문 횟수를 return하면 됩니다.

이를 바탕으로 BFS을 구현하면 아래와 같습니다.

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


def BFS(x, y):
    queue = deque([(x, y)])
    graph[y][x] = 0

    while queue:
        x, y = queue.popleft()

        if x == target[0] and y == target[1]:
            return graph[y][x]

        for i in range(8):
            next_x, next_y = x + dx[i], y + dy[i]

            if 0 <= next_x < L and 0 <= next_y < L:
                if graph[next_y][next_x] == 0:
                    queue.append([next_x, next_y])
                    graph[next_y][next_x] = graph[y][x] + 1

여기에 input을 받아서 탐색을 실시해주면 최종적인 아래의 코드로 구현이 완료됩니다.

 

Code

import sys
from collections import deque
input = sys.stdin.readline
T = int(input())

# 나이트의 이동 경우의 수
dx = [1, 2, 2, 1, -1, -2, -2, -1]
dy = [2, 1, -1, -2, -2, -1, 1, 2]


def BFS(x, y):
    queue = deque([(x, y)])
    graph[y][x] = 0

    while queue:
        x, y = queue.popleft()

        if x == target[0] and y == target[1]:
            return graph[y][x]

        for i in range(8):
            next_x, next_y = x + dx[i], y + dy[i]

            if 0 <= next_x < L and 0 <= next_y < L:
                if graph[next_y][next_x] == 0:
                    queue.append([next_x, next_y])
                    graph[next_y][next_x] = graph[y][x] + 1


for case in range(T):
    L = int(input())
    graph = [[0 for _ in range(L)] for _ in range(L)]
    knight = list(map(int, input().split()))
    target = list(map(int, input().split()))
    print(BFS(knight[0], knight[1]))
반응형