P3817 小A的糖果——贪心

小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$。

分析

1. 此题的贪心策略就是：优先吃后面盒子的糖果；因为吃后面盒子的糖果，能给后后面的盒子压力更小；
2. 此题的数据有一点水，可能只吃后面盒子，不够吃，不能达到数量t；按理说分情况讨论下，就是下面注释部分；但此题没卡这个测试点，可以不加；
3. 注意此题开long long

#include<bits/stdc++.h>

using namespace std;

int n, x;
long long ans;
int a[100010];

int main() {
    cin >> n >> x;
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
    }
    for (int i = 0; i < n - 1; i++) {
        int t = a[i] + a[i + 1] - x;//需要吃掉的糖果
        if (t > 0) {
            //优先吃后面盒子的糖果
            a[i + 1] -= t;
            /*
             * 可能只吃后面盒子，不够吃，不能达到数量t
            if(a[i+1]-t>=0){
                //后面盒子够吃
                a[i + 1] -= t;
            }else{
                //只吃后面盒子没有达到数量t
                a[i+1]=0;
                a[i]-=(t-a[i+1]);
            }*/
            ans += t;
        }
    }
    cout << ans;
    return 0;
}