蓝桥杯 算法训练 方格取数Python实现(动态规划)

资源限制
时间限制:1.0s 内存限制:256.0MB
问题描述
  设有N * N的方格图(N<=10),我们将其中的某些方格中填入正整数,而其他的方格中则放入数字0。
  某人从图的左上角的A 点(1,1)出发,可以向下行走,也可以向右走,直到到达右下角的B点(N,N)。在走过的路上,他可以取走方格中的数(取走后的方格中将变为数字0)。就是说路径不能重复
  此人从A点到B 点共走两次,试找出2条这样的路径,使得取得的数之和为最大。
输入格式
  输入的第一行为一个整数N(表示N*N的方格图),接下来的每行有三个整数,前两个表示位置,第三个数为该位置上所放的数。一行单独的0表示输入结束。
输出格式
  只需输出一个整数,表示2条路径上取得的最大的和。
样例输入
  8
  2 3 13
  2 6 6
  3 5 7
  4 4 14
  5 2 21
  5 6 4
  6 3 15
  7 2 14
  0 0 0
样例输出
  67

分析:
这个题目和之前做的最大值路径挺像的,只不过最大路径是一个人走,走出一条最大的路径就可以,但是这个题目相当于两个人在走,两个人的路径之和要最大,并且根据题中黄色字体,两个人的路径是不能重复的。我们还是按照二维动态规划的思路去考虑:

对于第一个人,假如动规数组是dp,那么
dp[i][j] = max(dp[i-1][j] , dp[i][j-1]) + s[i][j]
这个s是原来的数组,s[i][j]是表示i,j位置上的步数。

对于第二个人,同理可得:
dp[x][y] = max(dp[x-1][y] , dp[x][y-1]) + s[x][y]
同样s[x][y]是表示x,y位置上的步数。

我试着按照两个路线来做两次动态规划,理论上是可以的,但是介于两条路线不能重复,我们在动态规划时候取的max(dp[i-1][j] , dp[i][j-1]),并不知道是哪个角标的更大,就不方便将原来的数组的对应位置置零,所以放弃这种想法。

我们看两个人走路之间的关系,看是否能放到一起考虑:
在这里插入图片描述
所以我们可以写个四维的动态规划,在Python里面(不能导入别的模块的情况下),我们只能用列表嵌套列表嵌套列表嵌套列表,来表示我们想要的数据结构,(复杂度也是够大的,但是后面应该可以优化)

我们先给一个四维的数组,并全部置零:

    dp = [[[[0 for i in range(n+1)]for j in range(n+1)]for x 
  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值