路径之谜(1个超时)

题目描述
 

        小明冒充 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 = ' ')

    





评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值