CCF-小明种苹果续-python3

新增知识点：

        对

'''
N 表示苹果树个数
mi 表示该行的数字个数
ai 表示操作记录
    若为0或负数：表示去掉苹果是abs(ai)
    若为正数：表示重新统计了
求
T 剩下苹果总数
D 苹果掉落的苹果树个数
E 连续三次苹果掉落的苹果树个数
'''

n = int(input())  # 苹果树个数
t = 0  # 剩下苹果总数
d = 0  # 苹果掉落的苹果树个数
e = 0  # 连续三颗树有苹果掉落的组数
isDrop = []
for i in range(n):
    finput = input().split()
    m = int(finput[0])  # 该行的数字个数
    apple = list(map(int, finput[1:]))
    res = apple[0]  # 剩下苹果个数初始化为原来苹果个数
    drop = 0  # 是否发生苹果掉落
    for j in range(1, m):
        if apple[j] <= 0:
            res += apple[j]
        else:
            if apple[j] < res:  # 苹果掉落
                drop = 1
            res = apple[j]
    t += res
    if drop == 1:
        d += 1
    isDrop.append(drop)
# 判断isDrop里有几个连续3个为1，isDrop是一个‘圆圈’
for i in range(n):
    if isDrop[i] == 1 and isDrop[(i + 1) % n] == 1 and isDrop[(i + 2) % n] == 1:
        e += 1
print(t, d, e)

于循环数组的处理：巧妙运用下标