小A的糖果(c++题解)

题目描述

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

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

输入格式

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

第二行有 n 个用空格隔开的整数，第 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 颗。

错误代码：

#include <bits/stdc++.h>
using namespace std;
int a[1000005];
int n,m,ans;
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
    	scanf("%d",&a[i]);
	}
	for(int i=1;i<n;i++){
		if(a[i]+a[i+1]>m){ 如果相邻两个数，第一个数大于m，就把它吃到m个
			if(a[i]>m){
				ans+=a[i]-m;
				a[i]=m;
			}
			if(a[i]+a[i+1]>m){ 如果还大于m就把它第二堆吃到两个数总共还剩m个
				ans+=a[i]+a[i+1]-m;
			    a[i+1]=m-a[i];
			}
		}
	}
	cout<<ans;
}

_____________________________________________________________________________

呜呜呜，莫名其妙错两个测试点，还是去看了一下论坛，才发现要开longlong;

写作不易，点个赞呗!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! 

_____________________________________________________________________________

正确代码：

#include <bits/stdc++.h>
using namespace std;
int a[1000005];
long long n,m,ans;
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
    	scanf("%d",&a[i]);
	}
	for(int i=1;i<n;i++){
		if(a[i]+a[i+1]>m){如果相邻两个数，第一个数大于m，就把它吃到m个
			if(a[i]>m){
				ans+=a[i]-m;
				a[i]=m;
			}
			if(a[i]+a[i+1]>m){如果还大于m就把它第二堆吃到两个数总共还剩m个
				ans+=a[i]+a[i+1]-m;
			    a[i+1]=m-a[i];
			}
		}
	}
	cout<<ans;
}