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 盒糖吃掉 66 颗，第 4 盒吃掉 22 颗，第 6 盒吃掉 33 颗。

---

数据规模与约定

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

代码 

#include<bits/stdc++.h>

using namespace std;

long long ans=0;
long long a[100010];

int main(){
	long long n,x;
	scanf("%ld%ld",&n,&x);
	for(long i=0;i<n;i++) cin>>a[i];
	for(long i=0;i<n-1;i++){
		if(a[i]+a[i+1]>x){
			ans+=a[i+1]+a[i]-x;
            if(a[i+1]+a[i]-x>a[i+1]) a[i+1]=0;
            else a[i+1]=x-a[i];
		}
	}
	printf("%lld",ans);
	return 0;
}