【洛谷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

题解
贪心，每次使i与i-1比较，超过就从i拿，直到拿完再拿i-1。O（n）

代码

#include<bits/stdc++.h>
#define oo 100000000
#define ll long long
#define mo 1000000007
#define N 50005
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,x,a[100005];
ll ans;
int main()
{
    n=read();x=read();
    for (int i=1;i<=n;i++)
        a[i]=read();
    for (int i=1;i<=n;i++)
    {
        ll sum=(ll)a[i]+a[i-1];
        if (sum>x)
        {
            ans+=(ll)sum-x;
            if (sum-x>a[i]) a[i]=0;
            else a[i]=a[i]-(sum-x);
        }
    }
    printf("%lld",ans);
    return 0;
}