【YBT高效进阶】1基础算法/3二分算法/1数列分段 &【洛谷】P1182数列分段
题目描述
对于给定的一个长度为N的正整数数列 A 1至 AN,现要将其分成 M( M ≤ N M\leq N M≤N)段,并要求每段连续,且每段和的最大值最小。
关于最大值最小:
例如一数 4 2 4 5 1 要分成 33 段。
将其如下分段:
[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。
输入格式
第 1 行包含两个正整数N,M。
第 2 行包含 N 个空格隔开的非负整数 Ai ,含义如题目所述。
输出格式
一个正整数,即每段和最大值最小为多少。
输入输出样例
输入 #1复制
5 3
4 2 4 5 1
输出 #1复制
6
说明/提示
对于 20% 的数据, N ≤ 10 N \leq 10 N≤10。
对于 40% 的数据, N ≤ 1000 N\leq 1000 N≤1000。
对于 100% 的数据, 1 ≤ N ≤ 1 0 5 , M ≤ N , 1\leq N\leq 10^5,M\leq N, 1≤N≤105,M≤N,Ai <108, 答案不超过 109。
思路
二分枚举最大值,检测一次。
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
long long a[100010],n,m;
bool check(long long x)//检查是否符合条件
{
long long sum=0,cnt=1,i;
for(i=1;i<=n;i++)
{
if(sum+a[i]<=x)
sum+=a[i];
else
{
cnt++;
sum=a[i];
}
}
return (cnt<=m);
}
int main()
{
long long i,ans,l,r,mid;
scanf("%lld%lld",&n,&m);
for(i=1,l=r=0;i<=n;i++)
scanf("%lld",&a[i]),l=max(l,a[i]),r+=a[i];
while(l<r)//二分查找
{
mid=(l+r)>>1;
if(check(mid)) r=mid;
else l=mid+1;
}
printf("%lld",l);
return 0;
}