AcWing 1015. 摘花生 python3解法
原题链接
简单
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# author:ShidongDu time:2020/4/9
'''
Hello Kitty想摘点花生送给她喜欢的米老鼠。
她来到一片有网格状道路的矩形花生地(如下图),从西北角进去,东南角出来。
地里每个道路的交叉点上都有种着一株花生苗,上面有若干颗花生,经过一株花生苗就能摘走该它上面所有的花生。
Hello Kitty只能向东或向南走,不能向西或向北走。
问Hello Kitty最多能够摘到多少颗花生。
1.gif
输入格式
第一行是一个整数T,代表一共有多少组数据。
接下来是T组数据。
每组数据的第一行是两个整数,分别代表花生苗的行数R和列数 C。
每组数据的接下来R行数据,从北向南依次描述每行花生苗的情况。每行数据有C个整数,按从西向东的顺序描述了该行每株花生苗上的花生数目M。
输出格式
对每组输入数据,输出一行,内容为Hello Kitty能摘到得最多的花生颗数。
数据范围
1≤T≤100,
1≤R,C≤100,
0≤M≤1000
输入样例:
2
2 2
1 1
3 4
2 3
2 3 4
1 6 5
输出样例:
8
16
'''
# 首先,动态规划问题要注意两个点:1、不重 2、不漏
# 不重的话,不是必要条件,对于属性为min、max的题目可以忽略,对于属性为count的题目不可以忽略
# 不漏的话,这是必要条件
# 状态表示:集合:dp[i][j]表示从dp[1][1]走到dp[i][j]的所有路线 属性:max
# 状态计算:有两种情况:①从上面下来 ②从左边过来
# dp[i][j] = max(dp[i-1][j], dp[i][j-1]) + a[i][j]
from typing import List
class Solution:
def penats(self, group: List[List[int]]):
dp = [[0 for _ in range(len(group[0])+1)] for _ in range(len(group)+1)]
for i in range(1, len(dp)):
for j in range(1, len(dp[0])):
dp[i][j] = max(dp[i-1][j], dp[i][j-1]) + group[i-1][j-1]
return dp[-1][-1]
if __name__ == '__main__':
solution = Solution()
res = []
groups_num = int(input())
for i in range(groups_num):
row_num, col_num = list(map(int, input().split()))
cur_group = []
for j in range(row_num):
cur_group.append(list(map(int, input().split())))
tmp_res = solution.penats(cur_group)
res.append(tmp_res)
for k in res:
print(k)