【贪心】数列分段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。
#include<bits/stdc++.h>
using namespace std;
int a[100001];
int main()
{
int n,k,p,ans,sum;
while(cin>>n>>k)
{
ans=0;
sum=0;
memset(a,0,sizeof(a));
for(int i=1;i<=n;i++)
{
cin>>a[i];
if(a[i]>k)//如果输入的这个数比K还要大,sum不用加,ans++;
{
ans++;
if(sum)//如果再此之前sum还有值,sum清零,ans++;
ans++;
sum=0;
}
else
sum+=a[i];
if(sum<k&&i==n)
ans++;
if(sum>=k) //如果这一段的和大于等于k
{
ans++; //段数+1;
if(sum==k) //如果sum>=K,sum=0
sum=0;
else
{
sum=a[i]; //如果比K大,sum等于这个a[i]
if(i==n) //如果加上最后一段还比K大的话,最后一段单独成段
ans++;
}
}
}
cout<<ans<<endl;
}
return 0;
}