P3817 小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

思路:第一盒和最后一盒能不吃就不吃,因为吃了就只分别影响后、前的一盒。但是吃一盒中间的就影响 前后两盒。所以先判断第一盒,如果大于x,就吃到x为止,不要再贪吃了,ok,然后看第二盒让  第一盒+第二盒=x,第二盒为0,以此类推。如果第一盒小于x  但是第一盒+第二盒 >x,则吃第二盒直到  第一盒+第二盒 == x。这样就不用管最后一盒,可能你会举一个反例,比如:

3 4

7 5 1

这样的数据按照思路,就没法让相邻两盒为4了,但是我告诉你结果绝对没错,不信你自己允许一下程序看看是不是等于8,这并不是巧合,

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector> 
#include<cstdio>
#include<time.h> 
using namespace std;
int a[100001];
int n;
 
int main()
{
	cin>>n;
	int x; 
	cin>>x;
	long long sum=0;
	for(int i=0;i<n;i++)
	cin>>a[i];
	if(a[0]>x)
	{
		sum+=a[0]-x;
		a[0]=x;
	}
	
	
	for(int i=1;i<n;i++)
	{
		if(a[i-1]+a[i]>x)
		{
			sum+=(a[i-1]+a[i]-x);
			a[i]=x-a[i-1];
		}
	}
	cout<<sum<<endl;
	return 0;
	

}          
发布了44 篇原创文章 · 获赞 72 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/cj151525/article/details/90270663