ccf_201909-1 小明种苹果-python

题目以及测似案例

解决思路

输入实际是一个矩阵，行表示每颗树，第一列表示每棵树的苹果数，其他列表示每棵树对应的每次疏果操作。计算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)