贪心算法)"/>
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;}
更多推荐
P3817 小A的糖果(贪心算法)
发布评论