luogu3817 小A的糖果
题目描述
小A有N个糖果盒,第i个盒中有a[i]颗糖果。
小A每次可以从其中一盒糖果中吃掉一颗,他想知道,要让任意两个相邻的盒子中加起来都只有x颗或以下的糖果,至少得吃掉几颗糖。
输入输出格式
输入格式:
第一行输入N和x。
第二行N个整数,为a[i]。
输出格式:
至少要吃掉的糖果数量。
输入输出样例
输入样例#1:
3 3 2 2 2输出样例#1:
1输入样例#2:
6 1 1 6 1 2 0 4输出样例#2:
11输入样例#3:
5 9 3 1 4 1 5输出样例#3:
0说明
样例解释1
吃掉第二盒中的糖果。
样例解释2
第二盒吃掉6颗,第四盒吃掉2颗,第六盒吃掉3颗。
30%的测试数据,2<=N<=20,0<=a[i], x<=100
70%的测试数据,2<=N<=1000,0<=a[i], x<=10^5
100%的测试数据,2<=N<=10^5,0<=a[i], x<=10^9
将第一个糖果固定住,然后相邻的那块如果多了就删掉(并不会证明)。
#include <iostream> #include <algorithm> #include <cstdio> #include <cstring> #include <cstdlib> #include <map> #include <set> #include <vector> #include <queue> #include <cmath> using namespace std; const int N = 1e5 + 10; long long n, x, a[N], ans; int main() { // freopen("oi.in", "r", stdin); scanf("%lld%lld", &n, &x); for(int i = 1 ; i <= n ; i ++) { scanf("%lld", &a[i]); } for(int i = 1 ; i < n ; i ++) { if(a[i] + a[i + 1] > x) { ans += a[i] + a[i + 1] - x; a[i + 1] -= a[i] + a[i + 1] - x; } } printf("%lld\n", ans); }