数列分段(最小化最大值)

problem1

对于给定的一个长度为N的正整数数列A[i]，现要将其分成连续的若干段，并且每段和不超过M（可以等于M），问最少能将其分成多少段使得满足要求。

N≤100000，M≤10^9，M大于所有数的最小值，A[i]之和不超过10^9。

思路

由于段是连续的，可以贪心，从左往右扫描一遍，每一次超过m停止，将前面这一部分置为一段,统计段数。

时间复杂度$O(n)$

代码示例

//落谷 p1181
#include<bits/stdc++.h>
using namespace std;

const int maxn=100010;

int c[maxn];

int main()
{
    int n,m,Count;
    Count=1;
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;++i){
        scanf("%d",&c[i]);
    }
    int tmp=0;//阶段和
    for(int i=0;i<n;++i){
        tmp+=c[i];
        if(tmp>m){
            Count++;
            tmp=0;
            tmp+=c[i];
        }
    }
    printf("%d\n",Count);
    return 0;
}

problem2（p1的升级版）

给定一个长度为n的正整数序列，需要将其切成m段，要使每一段的和的最大值尽量小（最小化最大值）

例如一个数列 4 2 4 5 1 要求分成3段，每一段有一个和。问我怎么去分，能使得这三段和的最大值尽量小。

将其如下分段：

[4 2][4 5][1]

第一段和为6，第2段和为9，第3段和为1，和最大值为9。

将其如下分段：

[4][2 4][5 1]

第一段和为4，第2段和为6，第3段和为6，和最大值为6。

并且无论如何分段，最大值不会小于6。

所以可以得到要将数列4 2 4 5 1要分成3段，每段和的最大值最小为6。

N≤100000，M≤N， A[i]之和不超过10^9。

思路

枚举分段数m的话，对于每个分段确定最大值是$O(n)$的,因此总复杂度是$O(n*m)$的，不可行

假设最大的和值为max。那么当max变大时，分段数一定是减小的。

可以考虑反过来去枚举最值，这个最值会对应一个分段数（即problem1）。

由于上面提到的最值和段数之间的单调性，可以对最值区间二分。

当一个最值对应的段数等于题给m时，该最值即为ans

时间复杂度$O(n*log(suma_i))$

代码示例

//落谷 p1181
#include<bits/stdc++.h>
using namespace std;

const int maxn=100010;

int c[maxn];

int n,m;

int check(int mm)
{
    int Count=1;
    int tmp=0;//阶段和
    for(int i=0;i<n;++i){
        tmp+=c[i];
        if(tmp>mm){
            Count++;
            tmp=0;
            tmp+=c[i];
        }
    }
    return Count;
}

int main()
{
    scanf("%d %d",&n,&m);
    for(int i=0;i<n;++i){
        scanf("%d",&c[i]);
    }
    int Max=-1;
    for(int i=0;i<n;++i) Max=max(Max,c[i]);
    int l=Max-1,r=1e9+10;//注意下界
    while(r-l>1)
    {
        int mid=(l+r)/2;//向下取整
        if(check(mid)>m) l=mid;
        else r=mid;
    }
    cout<<r<<endl;//左开右闭
    return 0;
}