算法训练 跳马（python）

资源限制

时间限制：1.0s   内存限制：256.0MB

问题描述

　　一个8×8的棋盘上有一个马初始位置为(a,b)，他想跳到(c,d)，问是否可以？如果可以，最少要跳几步？

输入格式

　　一行四个数字a,b,c,d。

输出格式

　　如果跳不到，输出-1；否则输出最少跳到的步数。

样例输入

1 1 2 3

样例输出

1

数据规模和约定

　　0<a,b,c,d≤8且都是整数。

代码：（使用栈方法来保存坐标和步数）

a, b, c, d = map(int, input().split(' '))

step = [(1, 2), (1, -2), (-1, 2), (-1, -2), (2, 1), (2, -1), (-2, 1), (-2, -1)]

visit = [[False] * 8 for _ in range(8)]

queue = [(a, b, 0)]  # 坐标、步数

flag = False
while queue:
    n = len(queue)
    for i in range(n):
        y, x, t = queue.pop(0)

        if y == c and x == d:
            print(t)
            flag = True
            break

        for dy, dx in step:
            ny = y + dy
            nx = x + dx
            if 0 < ny < 9 and 0 < nx < 9 and not visit[ny-1][nx-1]:
                queue.append((ny, nx, t + 1))
                visit[ny-1][nx-1] = True

if not queue and not flag:
    print(-1)