每日一题（5）--贪心

小A的糖果

题目描述

小 A 有 $n$ 个糖果盒，第 $i$ 个盒中有 $a_i$ 颗糖果。

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

输入格式

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

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

输出格式

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

样例 #1

样例输入 #1

3 3
2 2 2

样例输出 #1

1

样例 #2

样例输入 #2

6 1
1 6 1 2 0 4

样例输出 #2

11

样例 #3

样例输入 #3

5 9
3 1 4 1 5

样例输出 #3

0

提示

样例输入输出 1 解释

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

---

样例输入输出 2 解释

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

---

数据规模与约定

• 对于 $30\%$ 的数据，保证 $n\le 20$，$a_i,x\le 100$。
• 对于 $70\%$ 的数据，保证 $n\le 10^3$，$a_i,x\le 10^5$。
• 对于 $100\%$ 的数据，保证 $2\le n\le 10^5$，$0\le a_i,x\le 10^9$。

基本思路

对于每一组，先看单个元素是否超过x，再相加看和是否超过x。首先考虑第一组的第一个元素，它只构成一组。如果第一个元素超过x，那么就将第一个元素减掉一部分。第二个元素同理。现在单个元素不会超过x，但如果两者的和超过x，应该在第二个元素上做减法。因为第二个元素还参与构成下一组，而第一个元素对其他组没有任何影响。接着是第二组，由于第一个元素已经处理过了，所以应该处理第二个元素。

解法

#include <iostream>
using namespace std;

int main () {
	int n, x;
	cin >> n >> x;
	long long ans = 0;
	int a[n];
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}
	if (a[0] > x) {
		ans += a[0] - x;
		a[0] = x;
	}
	for (int i = 1; i < n; i++) {
		if ((a[i] + a[i - 1]) > x) {
			ans += a[i] + a[i - 1] - x;
			a[i] -= a[i] + a[i - 1] - x;
		}
	}
	cout << ans;
	return 0;
}
```