P3817 小A的糖果

题目描述

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

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

输入格式

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

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

输出格式

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

输入输出样例

输入

3 3
2 2 2

输出 

1

输入

6 1
1 6 1 2 0 4

输出

11

输入 

5 9
3 1 4 1 5

输出 

0

说明/提示

样例输入输出 1 解释

吃掉第 2 盒中的一个糖果即可。

---

样例输入输出 2 解释

第 2 盒糖吃掉 66 颗，第 4 盒吃掉 22 颗，第 6 盒吃掉 33 颗。

---

数据规模与约定

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

思路

- 本题用到的算法，枚举判断

- 首先判断a[0]是否大于x，如果大于，则将a[0]=x，并将最后结果res+。

-当读入第二个数时，开始求和，若和大于x，则res+，a[i]改变，依次枚举求得结果。

代码

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;

const int N = 100010;
typedef long long ll; 

ll a[N];

int main() {
	ll n, x;
	cin >> n >> x;
	for (ll i = 0; i < n; i++)
		cin >> a[i];
	ll res = 0, b = a[0] ;
	for (ll i = 1; i < n; i++) {
		if (a[0] > x)
			res +=  a[0] - x, a[0] = x, b = a[0];
		if (b + a[i] > x) {
			res += (b + a[i]) - x;
			a[i] -= (b + a[i]) - x;
		}
		b = a[i];
	}
	cout << res << endl;
	return 0;
}