算法训练—拿金币

资源限制

时间限制：1.0s   内存限制：256.0MB

问题描述

　　有一个N x N的方格,每一个格子都有一些金币,只要站在格子里就能拿到里面的金币。你站在最左上角的格子里,每次可以从一个格子走到它右边或下边的格子里。请问如何走才能拿到最多的金币。

输入格式

　　第一行输入一个正整数n。
　　以下n行描述该方格。金币数保证是不超过1000的正整数。

输出格式

　　最多能拿金币数量。

样例输入

3
1 3 3
2 2 2
3 1 2

样例输出

11

数据规模和约定

　　n<=1000

代码1

n = int(input())
dp = [[]for i in range(n)]
for i in range(n):
  line = input().split( )
  for j in range(n):
      dp[i].append(int(line[j]))
#完成样例的输入
summ = [[0 for i in range(n)]for j in range(n)]
for i in range(n):
    for j in range(n):
        summ1 = summ[i][j - 1] + dp[i][j]
        summ2 = summ[i - 1][j] + dp[i][j]
        if summ1 >= summ2:
            summ[i][j] = summ1
        else:
            summ[i][j] = summ2
print(summ[n-1][n-1])

运行超时，70分。

代码2

n = int(input())
dp = [[]for i in range(n)]
for i in range(n):
  line = input().split( )
  for j in range(n):
      dp[i].append(int(line[j]))
if dp[0][n-1]>=dp[n-1][0]:
    cha = dp[0][n-1]
else:
    cha = dp[n-1][0]
for i in range(n):
    for j in range(n):
        if i==0and j!=0:
            dp[i][j]=dp[i][j - 1] + dp[i][j]
        elif j==0and i!=0:
            dp[i][j]=dp[i - 1][j] + dp[i][j]
        else:
            summ1 = dp[i][j - 1] + dp[i][j]
            summ2 = dp[i - 1][j] + dp[i][j]
            if summ1 >= summ2:
              dp[i][j] = summ1
            else:
              dp[i][j] = summ2
print(dp[n-1][n-1]-cha)

cha是python自我测试时发现正确答案和输出结果的差值而补全的。运行超时，90分，说明比原先复杂度低了一些，导致有些测试数据可以了。

代码3

n = int(input())
dp = [[]for i in range(n)]
for i in range(n):
  line = input().split( )
  for j in range(n):
      dp[i].append(int(line[j]))
#完成样例的输入
summ = [[0 for i in range(n)]for j in range(n)]
for i in range(n):
    for j in range(n):
        summ[i][j]=max(summ[i-1][j],summ[i][j-1])+dp[i][j]
print(summ[n-1][n-1])

用max函数代替if判断语句，简化代码，正确。

代码4

n = int(input())
dp = []
for i in range(n):
    dp.append(list(map(int, input().split())))
#完成样例的输入
summ = [[0 for i in range(n)]for j in range(n)]
for i in range(n):
    for j in range(n):
        summ[i][j]=max(summ[i-1][j],summ[i][j-1])+dp[i][j]
print(summ[n-1][n-1])

输入部分再次简化，正确。

小结：在路线规划时，最后一行只能右行，最后一列只能下行，这个问题在代码2考虑了，在其他似乎没有考虑，其实不然。因为代码2省略了summ，总和累积到各自的dp。而其他代码段均使用了summ，且初始赋值为0，避免了边的不省心。

小结后面的啰嗦：

代码5

n = int(input())
dp = [[]for i in range(n)]
for i in range(n):
  line = input().split( )
  for j in range(n):
      dp[i].append(int(line[j]))
for i in range(1, n):
    dp[i][0] = dp[i - 1][0] + dp[i][0]#往下边走
    dp[0][i] = dp[0][i - 1] + dp[0][i]#往右面走。初始状态
for i in range(1,n):
    for j in range(1,n):
        summ1 = dp[i][j - 1] + dp[i][j]
        summ2 = dp[i - 1][j] + dp[i][j]
        if summ1 >= summ2:
            dp[i][j] = summ1
        else:
            dp[i][j] = summ2
print(dp[n-1][n-1])

对代码2 的改进，正确。