路径之谜（py）

题目描述
小张冒充 X 星球的骑士，进入了一个奇怪的城堡。

城堡里边什么都没有，只有方形石头铺成的地面。

假设城堡地面是 n×n 个方格。如下图所示。

按习俗，骑士要从西北角走到东南角。可以横向或纵向移动，但不能斜着走，也不能跳跃。每走到一个新方格，就要向正北方和正西方各射一箭。（城堡的西墙和北墙内各有 n 个靶子）同一个方格只允许经过一次。但不必走完所有的方格。如果只给出靶子上箭的数目，你能推断出骑士的行走路线吗？有时是可以的，比如上图中的例子。

本题的要求就是已知箭靶数字，求骑士的行走路径（测试数据保证路径唯一）

输入描述
第一行一个整数 N (0≤N≤20)，表示地面有 N×N 个方格。

第二行 N 个整数，空格分开，表示北边的箭靶上的数字（自西向东）

第三行 N 个整数，空格分开，表示西边的箭靶上的数字（自北向南）

输出描述
输出一行若干个整数，表示骑士路径。

为了方便表示，我们约定每个小格子用一个数字代表，从西北角开始编号: 0,1,2,3 ⋯⋯

比如，上图中的方块编号为：

0 1 2 3

4 5 6 7

8 9 10 11

12 13 14 15

输入输出样例
示例

输入

4
2 4 3 4
4 3 3 3
输出

0 4 5 1 2 3 7 11 10 9 13 14 15
 

 思路：
        这道题要求寻找一条路线，并打印出路线，用dfs就很合适。
        但是这道题的难点我认为在于是如何去剪枝，除了必要的剪枝条件，这个条件也很重要：每走到一个格子，对应的靶子上数量减一，靶子上的数字为零时，就结束dfs，

参考代码：

# 定义一个函数，接受地面方格的大小n，北边箭靶的数字N，西边箭靶的数字W作为输入
def find_knight_path(n, N, W):
    # 创建一个二维数组，用于标记每个方格是否已经被访问过
    vis = [[True] * n for _ in range(n)]

    # 深度优先搜索函数，接受当前位置的横纵坐标(x, y)和已经走过的路径path作为输入
    def dfs(x, y, path):
        global ans, flag, step  # 声明全局变量

        # 更新当前位置的状态
        vis[x][y] = False  # 标记当前位置为已访问
        N[y] -= 1  # 更新北边箭靶的数字
        W[x] -= 1  # 更新西边箭靶的数字

        # 检查当前位置是否符合条件
        if N[y] < 0 or W[x] < 0:  # 如果箭靶的数字小于0，说明当前路径不符合条件，直接返回
            return
        elif x == y == n - 1:  # 如果到达了终点
            if len(path) == step:  # 如果路径长度等于总步数
                flag = True  # 找到了符合条件的路径
                ans = path  # 记录路径
        else:
            # 如果当前位置的箭靶数字已经减到0，但是仍有其他箭靶上的数字还没有减尽，则返回
            if N[y] == 0 and any(N[:y]):
                return
            if W[x] == 0 and any(W[:x]):
                return

        # 遍历四个方向
        for dx, dy in [(-1, 0), (0, 1), (1, 0), (0, -1)]:
            nx, ny = x + dx, y + dy  # 计算下一个位置的坐标
            if 0 <= nx < n and 0 <= ny < n and vis[nx][ny]:  # 检查下一个位置是否合法且未访问过
                dfs(nx, ny, path + [nx * n + ny])  # 递归调用DFS函数进行搜索
                if flag:  # 如果找到符合条件的路径，则直接返回
                    return
                vis[nx][ny] = True  # 回溯到上一个位置，标记为已访问
                N[ny] += 1  # 恢复北边箭靶的数字
                W[nx] += 1  # 恢复西边箭靶的数字


    # 调用DFS函数从起点开始搜索
    dfs(0, 0, [0])


# 读取输入
n = int(input())
N = list(map(int, input().split()))  # 读取北边箭靶上的数字
W = list(map(int, input().split()))  # 读取西边箭靶上的数字
step = sum(N)  # 总步数等于北边或者西边箭靶的数字之和
flag = False  # 是否找到路径的标志
ans = []  # 记录路径

# 调用函数求解路径并输出
find_knight_path(n, N, W)
# 输出结果
print(*ans)