首页 > 代码库 > luogu P3817 小A的糖果

luogu P3817 小A的糖果

题目描述

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

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

输入输出格式

输入格式:

 

第一行输入N和x。

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

 

输出格式:

 

至少要吃掉的糖果数量。

 

输入输出样例

输入样例#1:
3 32 2 2
输出样例#1:
1
输入样例#2:
6 11 6 1 2 0 4
输出样例#2:
11
输入样例#3:
5 93 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

贪心,每次贪后一个,不开long long 见祖宗

另外小a的牙应该坏掉了(捂脸)

 

#include <cstdio>#include <iostream>using namespace std;#define LL long longint a[500003];LL ans=0;int n,k;int main(){    scanf("%d%d",&n,&k);    for(int i=1;i<=n;i++)    {        scanf("%d",a+i);        if(a[i]>k)ans+=(a[i]-k),a[i]=k;    }    int tmp=0;    for(int i=1;i<n;i++)    {        if(a[i]+a[i+1]>k)        {            tmp=(a[i]+a[i+1]-k);            ans+=tmp;            a[i+1]-=tmp;        }    }    printf("%lld\n",ans);    return 0;}

 

luogu P3817 小A的糖果