数列分段II
时间限制: 1 Sec 内存限制: 128 MB
题目描述
对于给定的一个长度为N的正整数数列Ai,现要将其分成连续的若干段,并且每段和不超过M(可以等于M),问最少能将其分成多少段使得满足要求。
输入
第一行包含两个正整数N,M,表示了数列Ai的长度与每段和的最大值;
第二行包含N个空格隔开的非负整数Ai。
输出
仅包含一个正整数,输出最少划分的段数。
样例输入
复制样例数据
5 6
4 2 4 5 1
样例输出
3
提示
对于20%的数据,有N≤10;
对于40%的数据,有N≤1000;
对于100%的数据,有N≤105,M≤109,M大于所有数的最大值,Ai之和不超过109
思路:count计数,便利数组,将所有项加起来记为sum,如果sum>m了,就将i减一,往回退一个,将sum清零,count++。最后遍历完了之后i=n了,再加一次。
代码如下:
#include <cstdio>
#include <iostream>
#include <cmath>
#include <cstring>
typedef long long ll;
using namespace std;
ll a[100000];
int main()
{
int n,m;
int sum=0;
int count=0;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
for(int i=1;i<=n;i++)
{
sum+=a[i];
if(sum>m)
{
sum=0;
count++;
i--;
}
if(i==n)
{
count++;
}
}
printf("%d",count);
return 0;
}