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)