日常刷题笔记
题目描述
问题描述
有一个N x N的方格,每一个格子都有一些金币,只要站在格子里就能拿到里面的金币。你站在最左上角的格子里,每次可以从一个格子走到它右边或下边的格子里。请问如何走才能拿到最多的金币。
输入格式
第一行输入一个正整数n。
以下n行描述该方格。金币数保证是不超过1000的正整数。
输出格式
最多能拿金币数量。
样例输入
3
1 3 3
2 2 2
3 1 2
样例输出
11
思路
简单dp入门的思想哦!!!
拿到这个题作为新手的会想着去找状态转移方程。怎么转移呢?
大多数新手都是会去正向思维,我每一步是向右走还是向下走,这样走了之后怎么确定最大值呢?万一向右走了,却不是最优解,我该如何回溯呢?想着想着脑子就乱了。。满脸疑惑!!!
其实,入门级的dp通常需要逆向思维,即当前的最优解状态是如何从其他状态转移过来。
即 本题核心:
dp[i][j]=max(dp[i-1][j],dp[i][j-1])+coins[i][j]
当前的金币数(状态)只可能由上一格或者左一个的状态+当前格金币数组成,公式如上。好了,这题就写完了!
编码:两个for循环
for i in range(1,n):
for j in range(1,n):
dp[i][j]=max(dp[i-1][j],dp[i][j-1])+coins_list[i][j]
print(dp[n-1][n-1])
但是,会越界啊。 边界条件怎么办(第一行和第一列?
第一列的第i个状态 只可能由i-1状态转移过来
第一行的第j个状态 只可能由j-1状态转移过来
显然刚刚的代码不适合第一列
故,特殊情况特殊处理
for i in range(1,n):
dp[i][0]=dp[i-1][0]+coins_list[i][0]
dp[0][i]=dp[0][i-1]+coins_list[0][i]
完结!
代码
n=int(input())
coins_list=list()
for i in range(n):
tmp_list=list(map(int,input().split()))
coins_list.append(tmp_list)
dp=[[0 for i in range(n)] for j in range(n)]
dp[0][0]=coins_list[0][0]
for i in range(1,n):
dp[i][0]=dp[i-1][0]+coins_list[i][0]
dp[0][i]=dp[0][i-1]+coins_list[0][i]
for i in range(1,n):
for j in range(1,n):
dp[i][j]=max(dp[i-1][j],dp[i][j-1])+coins_list[i][j]
print(dp[n-1][n-1])