7562 나이트 이동

hey hey·2022년 4월 25일
0

알고리즘

목록 보기
44/57
post-thumbnail

문제

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

입력

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

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

출력

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

풀이

bfs를 이용해서 간단히 풀 수 있다.

  • heapq 를 이용해서 [횟수,y,x] 의 순서로 가장 횟수가 적은것부터 계산을 하여 계산횟수를 줄이고
  • 총 8방향으로 이동할 수 있게 하여 Q 에 추가해주는 방식을 이용했습니다.
import sys
sys.stdin = open('input.txt')
import heapq

T = int(input())

def bfs():
    Q = []
    heapq.heappush(Q,[0,now[1],now[0]])
    dy = [1,1,2,2,-1,-1,-2,-2]
    dx= [2,-2,1,-1,2,-2,1,-1]
    visited = [[False  for _ in range(L)] for _  in range(L)]
    while Q :
        cnt,y,x = heapq.heappop(Q)
        if y== goal[1] and x == goal[0]:
            print(cnt)
            break
        for i in range(8):
            ny = y + dy[i]
            nx = x + dx[i]
            if 0<=ny<L and 0<=nx<L:
                if visited[ny][nx]==False :
                    heapq.heappush(Q, [cnt+1,ny,nx])
                    visited[ny][nx]= True




for ts in range(T):
    L = int(input())
    now = list(map(int,input().split()))
    goal = list(map(int,input().split()))
    bfs()
profile
FE - devp

0개의 댓글