洛谷：P3817 小A的糖果

题目描述

小 A 有 n 个糖果盒，第 i 个盒中有 ai​ 颗糖果。

小 A 每次可以从其中一盒糖果中吃掉一颗，他想知道，要让任意两个相邻的盒子中糖的个数之和都不大于 x，至少得吃掉几颗糖。

输入格式

输入的第一行是两个用空格隔开的整数，代表糖果盒的个数 n 和给定的参数 x。

第二行有 n 个用空格隔开的整数，第 i 个整数代表第 i 盒糖的糖果个数 ai​。

输出格式

输出一行一个整数，代表最少要吃掉的糖果的数量。

输入输出样例

输入 #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 盒中的一个糖果即可。

样例输入输出 2 解释

第 2 盒糖吃掉 6 颗，第 4 盒吃掉 2 颗，第 6 盒吃掉 3 颗。

数据规模与约定

• 对于 30% 的数据，保证 n≤20，ai​,x≤100。
• 对于 70% 的数据，保证 n≤103，ai​,x≤105。
• 对于 100% 的数据，保证 2≤n≤105，0≤ai​,x≤109。

题解：

#include "iostream"
using namespace std;
#include "algorithm"
long long a[100005];
int main(){
int n;
long long x;
cin>>n>>x;
for(int i=0;i<n;i++){
cin>>a[i];
}

long long sum=0;//统计需要吃的总数
for(int i=1;i<n;i++){
long long num=a[i-1]+a[i];
if(num>x){
long long temp=num-x;//吃掉的数量
//先吃后盒子
long long t1=min(a[i],temp);
a[i]-=t1;
sum+=t1;
//吃前面盒子
long long t2=temp-t1;
a[i-1]-=t2;
sum+=t2;
}
}
cout<<sum<<endl;
system("pause");
return 0;
}