洛谷 小A的糖果 贪心问题

题目描述

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

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

输入格式

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

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

输出格式

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

根据贪心原则，单个超过x，必须吃到x，和超过x，吃掉下一个。

代码：

#include
#define MX 1000005
using namespace std;
int n,x;
int a[MX];
int main() {
    cin>>n>>x;
    long long int sum = 0;
    for(int i = 1;i <= n;i++)
    {
        cin>>a[i];
    }
    for(int i = 1;i <= n;i++)
    {
        if(a[i] > x)
        {
            sum = sum + a[i] - x;
            a[i] = x;
        }
        if(a[i] + a[i+1] > x)
        {
            sum = sum + a[i] + a[i+1] - x;
            a[i+1] = a[i+1] - (a[i] + a[i+1] - x);
        }
    }
    cout<     return 0;
}