问题描述
有一个N x N的方格,每一个格子都有一些金币,只要站在格子里就能拿到里面的金币。你站在最左上角的格子里,每次可以从一个格子走到它右边或下边的格子里。请问如何走才能拿到最多的金币。
输入格式
第一行输入一个正整数n。
以下n行描述该方格。金币数保证是不超过1000的正整数。输出格式
最多能拿金币数量。
样例输入
3
1 3 3
2 2 2
3 1 2样例输出
11
数据规模和约定
n<=1000
解题思路:
一开始想的是正向推导,但是发现在比较完之后的位置更迭有点不太好把握,然后就去网上找了找,发现可以反着来想:
用更新二维数组的方式,首先创建并初始化二维数组:dp[n][n] (n为输入的数据) 。选择使用dp[i][j]来存储更新的数据,直接覆盖之前的数据,比如示例中的dp[0][0]位置,下一步应该是dp[0][1] (因为dp[0][0]位置的右侧数据比下方数据大),那么就可以用dp[i][j]来存储当前的dp[0][1]的数据。这样也就可以知道 dp[i][j] 的值其实是由上一步的值推导出来的,具体可以分为三种:
- 当n=1时,第一步就是最后需要的数据:
此时i==0 and j==0 dp[i][j]=dp[i][j] (写这一句是当n != 1 时,循环也是要从 0 开始的,所以数据还是要存储的)
- 当dp[i][j]在上边界或者在左边界时:
也就是当 i==0 or j==0时;
当 i==0 时,能走到这步,肯定是从上个元素向下走的,那么此时的计算公式为:
dp[i][j]= dp[i-1][j]+dp[i][j]
(加上dp[i][j]是为了加上现在的值,才是总的和)
当j==0 时,能走到这步,肯定是从上个元素向右走的,那么此时的计算公式为:
dp[i][j] = dp[i][j-1]+dp[i][j]
- 当dp[i][j]在数组为中间位置时,此时就是取其上方和左方的最大值来相加:
dp[i][j]=dp[i][j]+max(dp[i-1][j]+dp[i][j-1])
代码实现:
n=int(input())
dp=[[]for j in range(n)]
for i in range(n):
#dp[i]=input().split()#这个样子输入也是可以输入成二维数组的,但是输入进去都是字符类型,
#而下面的计算里的 + 对于字符则是连接操作,而不是计算,想要转换类型还是用map
dp[i]=list(map(int,input().split()))#将每一个都转换成int类型,用list保存成数组形式
for i in range(n):
for j in range(n):
if i==0 and j==0:
dp[i][j] =dp[i][j]
elif i==0:
dp[i][j]=dp[i][j]+dp[i][j-1]
elif j==0:
dp[i][j]=dp[i][j]+dp[i-1][j]
else:
dp[i][j]=dp[i][j]+max(dp[i-1][j],dp[i][j-1])
print(dp[i][j])#这里可以用i j,因为里面的i j可以外带输出,但是仅限于输出数据,
#如果要在两个for循环外,使用dp[i][j]进行元素计算或者对比,那么i j 还是超出范围的
#print(dp[n-1][n-1])#意思和上面的print一样,因为最后i j循环到最后也是都是n-1
知识补充:
二维数组的元素由外界输入:
n=int(input())#n从外界输入,下面生成n*n的二维数组
dp=[[]for j in range(n)]
for i in range(n):
dp[i]=input().split()
以上 这个样子输入也是可以输入成二维数组的,但是输入进去都是字符类型,所以不能用 + 进行计算,只能表示字符连接
n=int(input())#n从外界输入,下面生成n*n的二维数组
dp=[[]for j in range(n)]
for i in range(n):
dp[i]=list(map(int,input().split()))
以上将每一个都转换成int类型,用list保存成数组形式