C++贪心——糖果

题目描述

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

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

输入格式

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

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

输出格式

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

分析：

先判断前一个数是否大于x，如果大于x，则将这个数等于这个数减去x，后面的数就等于0，吃掉的糖果等于第一个数减去x的值；如果不大于x，则判断两个数相加是否大于x，如果大于，吃掉的糖果数等于两个数相加的和减去x，第二个数等于x-第一个数。（第二个数应该等于这个数减去吃掉糖果的数量，吃掉的糖果数等于两个数相加的和减去x，整理完就是x减去第一个数值。）

代码：

#include<bits/stdc++.h>
using namespace std;
int a[1000001];
int  main() {
	long long n, x, j, sum = 0;
	cin >> n >> x;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	for (int i = 1; i < n; i++) {
		j = i + 1;
		if (a[i] > x) {//第一个数大于x时
			sum += a[i] - x;
			a[i] = a[i] - x;
			sum += a[j];
			a[j] = 0;
		}
		else {
			if (a[i] + a[j] > x) {//两个数相加大于x时
				sum += a[j] + a[i] - x;
				a[j]= x-a[i];
			}
		}
	}
	cout << sum;
	return 0;
}