luogu P1181 数列分段Section I x

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是最少划分的段数。

思路:

　　用now来记录当前段数所形成的和,如果大于最大的数m,则更新ans(即ans++),并且将now更新为当前输入的数(a[i]),可以手动模拟一下~

　　然后就是特殊情况(即now==m),因为刚好能够构成一段所以跟前面组成一段,所以将now置为0,ans++.

　　然后又因为输出的是段数,刚才的做法统计的是隔板的数目,所以在最后一块的时候需要判断是恰好组成一段(即now==m),还是又多了一段(即now<m),因为now<m的情况我们并没有进行考虑,所以加个特判:如果i==n && now==m 时ans++.

　　输出ans,程序结束.

坑点:

　　1)要看好输出什么

　　2)要开long long 

代码:

#include <iostream>
#include <cstdio>
#define LL long long

using namespace std;

const int M = 1e5 + 1;
LL n,m,now,ans;
LL a[M];

int main()
{
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        now+=a[i];
        if(now>m)
        {
            now=a[i];
            ans++;
        }
        else
        if(now==m)
        {
            now=0;
            ans++;
        }
        else
        if(now<m && i==n)///特判
        {
            ans++;
        }
    }
    printf("%lld",ans);
    return 0;
}

End.