P3817 小A的糖果(模拟+贪心)

题目描述

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

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

输入格式

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

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

输出格式

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

输入输出样例

输入 #1                                                输出 #1

3 3                               1
2 2 2                                 

输入 #2                                                 输出 #2

6 1                                                        11

1 6 1 2 0 4

输入 #3                                                  输出 #3

5 9                                                         0

3 1 4 1 5

说明/提示

样例输入输出 1 解释

吃掉第 2 盒中的一个糖果即可。

---

样例输入输出 2 解释

第 2 盒糖吃掉 66 颗，第 4 盒吃掉 22 颗，第 6 盒吃掉 33 颗。

---

数据规模与约定

• 对于 30%30% 的数据，保证 n≤20，ai​,x≤100。
• 对于 70%70% 的数据，保证 n≤10^3，ai​,x≤10^5。
• 对于 100%100% 的数据，保证 2≤n≤10^5，0≤ai​,x≤10^9。

代码题解：

#include<bits/stdc++.h>
using namespace std;
#define N 1000

int is_A_Candy(int n,int x,int a[])
{ 
    long long sum=0;//计数:表吃的糖数 
    if(a[1]>x)
    {
        sum+=a[1]-x;//当第一个盒子内的糖数大于参数X时,所吃的糖数
        a[1]=x;//a[i]>=x,要吃的最少，即是a[i]=x;
    }
    for(int i=2;i<=n;i++)
    {
        if(a[i]+a[i-1]>x)
        {
            sum+=a[i]+a[i-1]-x;//当两个盒子内的糖数大于参数X时,所吃的糖数
            a[i]=x-a[i-1]; 
        }
    }
    return sum;
}

int main()
{
    int i,n,x,a[N],count;
    cin>>n>>x;
    for(i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    count=is_A_Candy(n,x,a);
    cout<<count;
}