主要就是注意统计连续三棵树都发生掉落的数量时要考虑和最后俩颗树连续的是第一颗和第二棵树,其他没什么,代码中都有注释。
测试数据:
4
4 74 -7 -12 -5
5 73 -8 -6 59 -4
5 76 -5 -10 60 -2
5 80 -6 -15 59 0
N = int(input())
# 剩余苹果总数
T = 0
# 发生掉落的苹果树的数量
D = 0
# 连续三棵树发生掉落的数量
E = 0
# 用来记录每颗苹果树是否发生掉落
record = []
for _ in range(N):
temp = list(map(int,input().split()))
m = temp[0]
# 存放当前树的苹果总数
temp_sum = temp[1]
# 记录当前苹果是是否发生掉落
flag = False
for i in range(2,m+1):
# 说明是计数的操作
if temp[i] > 0:
# 当前苹果数量和之前记录的数量不一致,说明发生了掉落
if temp_sum != temp[i]:
flag = True
# 将苹果数量改为当前统计的数量
temp_sum = temp[i]
# 说明是疏果的操作
else:
temp_sum += temp[i]
if flag:
record.append(1)
D += 1
else:
record.append(0)
T += temp_sum
# 统计连续三棵树发生掉落的数量
for i in range(N):
# 记得取余,因为题目说苹果树是种的一个环
if record[i] == 1 and record[(i+1)%N] == 1 and record[(i+2)%N] == 1:
E += 1
print('{} {} {}'.format(T,D,E))