Fork me on GitHub

P3817 小A的糖果

luogu 传送门

挺好想的贪心(蒟蒻一下就想到啦)。
我们从前往后处理时,尽量吃后面的糖,因为后面的糖会对后面产生影响,而前面的就不会影响了。
不过要开long long ,为此我wa了两个点。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
int n,x;
LL a[100009],ans;
int main()
{
    scanf("%d%d",&n,&x);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    for(int i=1;i<=n;i++)
    {
        int p;
        if(a[i]>x) ans+=a[i]-x,a[i]=x;
        if(a[i]+a[i-1]>x)
        {
            p=a[i]+a[i-1]-x;
            if(a[i]>=p) a[i]-=p;
            else p-=a[i],a[i]=0,a[i-1]-=p;
            ans+=p;
        }
        if(a[i]+a[i+1]>x)
        {
            p=a[i]+a[i+1]-x;
            if(a[i+1]>=p) a[i+1]-=p;
            else p-=a[i+1],a[i+1]=0,a[i]-=p;
            ans+=p;
        }//注意两个if的顺序,先处理i和i-1,再处理i和i+1,因为前者会减小对后者的压力
    }
    printf("%lld",ans);
    return 0;
}
posted @ 2017-09-24 17:48  primes  阅读(132)  评论(0编辑  收藏  举报