每日一题
####P1182 数列分段 Section II
题目描述
对于给定的一个长度为N的正整数数列 A_{1\sim N}A1∼N,现要将其分成 MM(M\leq NM≤N)段,并要求每段连续,且每段和的最大值最小。
关于最大值最小:
例如一数列4 2 4 5 1 要分成 3 段。
将其如下分段:
4 2\4 5\1
第一段和为6,第 22 段和为 9,第 33 段和为 1,和最大值为 9。
将其如下分段:
4\2 4\ 5 1
第一段和为 4,第 2 段和为 6,第 3 段和为 6,和最大值为 6。
并且无论如何分段,最大值不会小于 6。
所以可以得到要将数列 4 2 4 5 1 要分成 3 段,每段和的最大值最小为 6。
输入格式
第 11 行包含两个正整数 N,M。
第 22 行包含 N个空格隔开的非负整数 Ai,含义如题目所述。
输出格式
一个正整数,即每段和最大值最小为多少。
输入输出样例
输入 #1
5 3
4 2 4 5 1
输出 #1
6
说明/提示
对于 20% 的数据,N≤10。
对于 40% 的数据,N≤1000。
对于 100% 的数据,N≤105,M≤N,Ai < 10^8, 答案不超过 10^9。
题解
#include <bits/stdc++.h>
using namespace std;
int n,m;
int a[100005];
int f(int x){//判断是否每一段都小于mid
int num = 1;
int sum = 0;
for(int i = 0;i < n;i++){
if(sum + a[i] <= x)//若小于,添加
sum += a[i];
else{//否则另开一段
sum = a[i];
num++;
}
}
if(num <= m)
return 1;
return 0;
}
int main(){
cin>> n >> m ;
int l = 0,r = 0;
int mid;
for(int i = 0;i < n;i++){
scanf("%d",&a[i]);
l = max(l,a[i]);
r += a[i];
}
while(l <= r){//二分
mid = (l + r) / 2;
if(f(mid))//mid大了,r左移
r = mid - 1;
else//小了
l = mid + 1;
}
cout << l << endl;
return 0;
}