题目以及测似案例
解决思路
输入实际是一个矩阵,行表示每颗树,第一列表示每棵树的苹果数,其他列表示每棵树对应的每次疏果操作。计算T 可以直接把每个输入数加起来即可。每次处理一行数据,计算疏果个数,存储对应苹果树编号。当后一行(既下一颗树)的疏果个数大于当前疏果个数时,更新K P
代码
str = input()
nums = str.rsplit()
nums = list(map(int, nums))
T = 0 # 最后树上剩下的苹果
K = 0 # 疏果个数最多的苹果树编号
P = 0 # 该苹果树的疏果个数
for N in range(1, nums[0]+1):
row = input()
row = row.rsplit()
row = list(map(int, row))
T += row[0]
sg_num = 0
for i in range(1, nums[1]+1):
sg_num += row[i]
T += sg_num
if P == 0:
P = 0 - sg_num
K = N
elif P < 0 - sg_num:
P = 0 - sg_num
K = N
print(T, K, P)