7562번 나이트의 이동 파이썬

문제

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

입력

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

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

출력

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

"""
알고리즘: BFS
최소 이동횟수를 구하는 것이기 때문에 DFS보다는 BFS를 사용
이동할때마다, B 배열에 (방문처리 기능도 함) 그 이전 value값을 다음 index value값에 넣기
"""
from collections import deque
#나이트의 이동방식
dx = [2,1,-1,-2,2,1,-1,-2]
dy = [1,2,2,1,-1,-2,-2,-1]

T = int(input())

def BFS(i,j):
    # 시작점과 종점이 같으면 바로 return
    if i == x2 and j ==y2:
        print(0)
        return

    queue = deque()
    queue.append([i,j])

    while queue:
        x,y = queue.popleft()
        for i in range(8):
            nx = x+dx[i]
            ny = y+dy[i]
            if -1<nx<size and -1<ny<size and B[nx][ny]==0:
                queue.append([nx,ny])
                B[nx][ny] = B[x][y]+1
                #끝나는 조건
                if nx == x2 and ny == y2:
                    print(B[nx][ny])
                    return

for i in range(T):
    # 데이터 받기
    size = int(input())
    x1,y1 = map(int,input().split())
    x2,y2 = map(int,input().split())
    # 보드 만들기
    B = [[0 for _ in range(size)] for _ in range(size)]

    BFS(x1,y1)

개인적으로, 그래프 탐색 문제는 요구하는 바가 더 잘 들어나서 수학, 구현 문제보다 알고리즘 아이디어를 떠올리기가 훨씬 수월하다.

술술 코드가 짜였으나, 방향 벡터 한 쌍의 값을 실수로 중복되게 설정해서 테스트케이스는 통과하지만, '틀렸습니다'가 떴었다.

질문검색에서 나랑 똑같은 사람이 있어서 그거 보고 해결했다.
하필 테스트케이스는 통과했기 때문에, 만약 실전이었다면 많이 헤맸을 것 같다.

좋은 웹페이지 즐겨찾기