P1181 数列分段Section I
题目描述
对于给定的一个长度为N的正整数数列A[i],现要将其分成连续的若干段,并且每段和不超过M(可以等于M),问最少能将其分成多少段使得满足要求。
输入输出格式
输入格式:输入文件divide_a.in的第1行包含两个正整数N,M,表示了数列A[i]的长度与每段和的最大值,第2行包含N个空格隔开的非负整数A[i],如题目所述。
输出文件divide_a.out仅包含一个正整数,输出最少划分的段数。
输入输出样例
输入样例#1:
5 6 4 2 4 5 1
输出样例#1:
3
说明
对于20%的数据,有N≤10;
对于40%的数据,有N≤1000;
对于100%的数据,有N≤100000,M≤10^9,M大于所有数的最小值,A[i]之和不超过109。
将数列如下划分:
[4][2 4][5 1]
第一段和为4,第2段和为6,第3段和为6均满足和不超过M=6,并可以证明3是最少划分的段数。
题解:从开始到最后模拟即可,每次下一个数不能放进去的时候就进行下一段。
AC代码:
#include<iostream>
using namespace std;
#define _for(i,a,b) for(int i=a;i<=b;i++)
const int maxn = 100007;
int n,m,a[maxn];
int main()
{
cin>>n>>m;
_for(i,1,n)cin>>a[i];
long long sum = 0;
int ans =0;
int ok = 0;
_for(i,1,n)
{
sum+=a[i];
if(sum>m)
{
sum = 0;
ans++;
i--;
if(i==n)ok = 1;
}
}
if(ok==0)ans++;
cout<<ans<<endl;
return 0;
}