题目描述
小明冒充 X 星球的骑士,进入了一个奇怪的城堡。城堡里边什么都没有,只有方形石头铺成的地面。假设城堡地面是 nxn 个方格。如下图所示。
按习俗,骑士要从西北角走到东南角。可以横向或纵向移动,但不能斜着走,也不能跳跃。每走到一个新方格,就要向正北方和正西方各射一箭。(城堡的西墙和北墙内各有N个靶子)同一个方格只允许经过一次。但不必走完所有的方格。如果只给出靶子上箭的数目,你能推断出骑士的行走路线吗?有时是可以的,比如上图中的例子。
本题的要求就是已知箭靶数字,求骑士的行走路径(测试数据保证路径唯一)
输入描述
第一行一个整数 N(0 < N < 20),表示地面有N x 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
n = int(input())
col = list(map(int,input().split())) #北边数字
row = list(map(int,input().split())) #西边数字
ans = []
step = (sum(col)+sum(row)) // 2 #路径总长
#初始化访问数组与路径字符串
on_path = [[False] * (n+1) for _ in range(n+1)] #记录每个方格是否被走过,走过为True
path = []
on_path[0][0] = True
# 定义移动方向,上下左右
dirs = [[0, -1], [0, 1], [-1, 0], [1, 0]]
def dfs(x,y):
#标记当前格已被访问
on_path[x][y] = True
row[x] -= 1
col[y] -= 1
path.append((x,y))
#边界条件,如果到达右下角,则成功
if x == n-1 and y == n-1 and sum(row) == 0 and sum(col) == 0:
return True
else:
#遍历当前单元格的四个方向:右左上下
for nx,ny in ((x+1,y),(x-1,y),(x,y+1),(x,y-1)):
#剪枝1:已访问过
if on_path[nx][ny] :
continue
#剪枝2:超出网格线
if nx < 0 or nx == n or ny < 0 or ny == n:
continue
#剪枝3:箭靶数字为0
if row[nx] == 0 or col[ny] == 0:
continue
if dfs(nx,ny):
return path
#如果所有方向都没找到有效路劲,则回溯,恢复现场
path.pop()
row[x] += 1
col[y] += 1
on_path[x][y] = False
dfs(0,0)
#打印出找到的路径
for x,y in path:
print(n*x+y,end = ' ')