洛谷_P3817 小A的糖果_python写法

P3817 小A的糖果 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

一开始有两个案例无法通过，然后看了一下测试案例发现其实是没有考虑到为0这个情况，因为如果为0那么无法吃掉任何一粒糖果，而按照之前的写法是到负数了，那后面的一个盒子的糖果数量加上一个负数是有可能不超过给定的参数，所以一定要加上一个判断是否为0 的判断条件。 

n, x = map(int,input().split())
data = list(map(int,input().split()))

ans = 0
for i in range(1,n):
    if data[i]+data[i-1]>x:
        ans += data[i]+data[i-1]-x
        if data[i] != 0:
            data[i] = data[i]-(data[i]+data[i-1]-x)
        else:
            data[i-1] = data[i-1] - (data[i] + data[i - 1] - x)

print(ans)