【洛谷 P3817】小A的糖果 题解(贪心+向量)
小A的糖果
题目描述
小 A 有 nnn 个糖果盒,第 iii 个盒中有 aia_iai 颗糖果。
小 A 每次可以从其中一盒糖果中吃掉一颗,他想知道,要让任意两个相邻的盒子中糖的个数之和都不大于 xxx,至少得吃掉几颗糖。
输入格式
输入的第一行是两个用空格隔开的整数,代表糖果盒的个数 nnn 和给定的参数 xxx。
第二行有 nnn 个用空格隔开的整数,第 iii 个整数代表第 iii 盒糖的糖果个数 aia_iai。
输出格式
输出一行一个整数,代表最少要吃掉的糖果的数量。
样例 #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 盒糖吃掉 666 颗,第 4 盒吃掉 222 颗,第 6 盒吃掉 333 颗。
数据规模与约定
- 对于 30%30\\%30% 的数据,保证 n≤20n \\leq 20n≤20,ai,x≤100a_i, x \\leq 100ai,x≤100。
- 对于 70%70\\%70% 的数据,保证 n≤103n \\leq 10^3n≤103,ai,x≤105a_i, x \\leq 10^5ai,x≤105。
- 对于 100%100\\%100% 的数据,保证 2≤n≤1052 \\leq n \\leq 10^52≤n≤105,0≤ai,x≤1090 \\leq a_i, x \\leq 10^90≤ai,x≤109。
思路
优先吃后面一盒。如果吃第一盒的话,不会影响第二盒和第三盒的和。
AC代码
#include <iostream>
#include <vector>
#define AUTHOR "HEX9CF"
using namespace std;int main() {int n, x;long long cnt = 0;vector<int> v;cin >> n >> x;for(int i = 0; i < n; i++) {int in;cin >> in;v.push_back(in);}vector<int>::iterator it = v.begin();for(; it + 1 != v.end(); it++) {int sum = *(it + 1) + *it;if(sum > x) {*(it + 1) -= sum - x;cnt += sum - x;}}cout << cnt << endl;return 0;
}