洛谷[P3817 小A的糖果] {贪心}
题目描述
小 A 有 n 个糖果盒,第 i 个盒中有ai颗糖果。
小 A 每次可以从其中一盒糖果中吃掉一颗,他想知道,要让任意两个相邻的盒子中糖的个数之和都不大于 x,至少得吃掉几颗糖。
输入格式
输入的第一行是两个用空格隔开的整数,代表糖果盒的个数 n 和给定的参数 x。
第二行有 nn 个用空格隔开的整数,第 i 个整数代表第 i 盒糖的糖果个数 ai 。
输出格式
输出一行一个整数,代表最少要吃掉的糖果的数量。
输入输出样例
输入 #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 盒中的两个糖果即可。
样例输入输出 2 解释
第 2 盒糖吃掉 6颗,第 4 盒吃掉 2 颗,第 6 盒吃掉 3 颗。
数据规模与约定
对于 30% 的数据,保证n≤20,ai,x≤100。
对于 70%70% 的数据,保证 n≤
1
0
3
10^3
103, ai,x≤
1
0
5
10^5
105。
对于 100%100% 的数据,保证2≤n≤
1
0
5
10^5
105,0≤ai ,x≤
1
0
9
10^9
109。
解题思路
这道题很容易被说明提示误导,会误认为只要处理偶数盒就可以实现题目要求,但是实际上这道题是应该分为两部分进行处理的。
第一盒糖果只会影响第二盒糖果,而第二盒之后的糖果会影响其相邻的两盒糖果。
因此,我们先将第一盒糖果进行处理,之后的糖果就可以使用循环处理。
完整代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
ll a[maxn],x;
int n;
int main()
{
scanf("%d %lld",&n,&x);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
ll ans=0;
if(a[1]>x)//单独考虑第一个
{
ans+=a[1]-x;
a[1]=x;//注意此时将a[1]的值赋为x
}
for(int i=1;i<=n;i++)//相当于从前往后推
{
if(a[i]+a[i+1]>x)//此时已经可以保证a[i]一定小于等于x
{
ans+=a[i]+a[i+1]-x;//ans表示多出来的部分和
a[i+1]=x-a[i];// a[i+1]=a[i+1]-( a[i]+a[i+1]-x) ,因为a[i]一定小于等于x
}
}
printf("%lld",ans);
return 0;
}