P3817 小A的糖果

题目描述

小A有N个糖果盒，第i个盒中有a[i]颗糖果。

小A每次可以从其中一盒糖果中吃掉一颗，他想知道，要让任意两个相邻的盒子中加起来都只有x颗或以下的糖果，至少得吃掉几颗糖。

输入输出格式

输入格式：

第一行输入N和x。

第二行N个整数，为a[i]。

输出格式：

至少要吃掉的糖果数量。

输入输出样例

输入样例#1：

3 3
2 2 2

输出样例#1：

1

输入样例#2：

6 1
1 6 1 2 0 4

输出样例#2：

11

输入样例#3：

5 9
3 1 4 1 5

输出样例#3：

0

说明

样例解释1

吃掉第二盒中的糖果。

样例解释2

第二盒吃掉6颗，第四盒吃掉2颗，第六盒吃掉3颗。

30%的测试数据，2<=N<=20，0<=a[i], x<=100

70%的测试数据，2<=N<=1000，0<=a[i], x<=10^5

100%的测试数据，2<=N<=10^5，0<=a[i], x<=10^9

标程

mmp

一开始

p=p-a[i+1];
a[i+1]=0;

写反了，所以每次p都相当于不减，，

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<algorithm>
#define lli long long int 
using namespace std;
void read(lli &n)
{
    char c='+';lli x=0;bool flag=0;
    while(c<'0'||c>'9')
    {c=getchar();if(c=='-')flag=1;}
    while(c>='0'&&c<='9')
    {x=x*10+(c-48);c=getchar();}
    flag==1?n=-x:n=x;
}
lli a[100001];
lli n,m;
lli ans=0;
int main()
{
    read(n);read(m);
    for(lli i=1;i<=n;i++)
        read(a[i]);
    for(lli i=1;i<=n;i++)
    {
        if(a[i]+a[i+1]>m)
        {
            lli p=a[i]+a[i+1];
            if(a[i+1]>=(p-m))
            {
                a[i+1]-=(p-m);
                ans+=p-m;
            }
            else 
            {
                ans+=a[i+1];
                p=p-a[i+1];
                a[i+1]=0;
                a[i]-=(p-m);
                ans+=(p-m);
            }
        }
    }
    printf("%lld",ans);
    return 0;
}