> 文章列表 > 【洛谷 P3817】小A的糖果 题解(贪心+向量)

【洛谷 P3817】小A的糖果 题解(贪心+向量)

【洛谷 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 20n20ai,x≤100a_i, x \\leq 100ai,x100
  • 对于 70%70\\%70% 的数据,保证 n≤103n \\leq 10^3n103ai,x≤105a_i, x \\leq 10^5ai,x105
  • 对于 100%100\\%100% 的数据,保证 2≤n≤1052 \\leq n \\leq 10^52n1050≤ai,x≤1090 \\leq a_i, x \\leq 10^90ai,x109

思路

优先吃后面一盒。如果吃第一盒的话,不会影响第二盒和第三盒的和。

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;
}