洛谷 P1182数列分段题解--zhengjun
思路
首先,看到题目“\(每段和最大值最小为多少\)”,就知道是二分答案了,我们可以二分“每段和最大值”,再来判断是否合法,就可以了。
在判断时,应该用一个变量保存当前加到了多少,如果再加上一个就大于当前mid, 就将组数加一,只要组数小了,就把right移到mid,(当前mid大了),否则,就把left移到mid。
代码实现
#include<bits/stdc++.h>
using namespace std;
int n,m,a[100005],maxx=0;
int check(int mid) {
if(mid<maxx)return 0;//如果mid比数列中最小的数还小,就直接过滤掉
int sum=0,k=0;
for(int i=1; i<=n; i++) {
if(sum+a[i]>mid) {
k++;
sum=a[i];//清空后,加上a[i]
} else sum+=a[i];
}
if(k>=m)return 0;
return 1;//组数太少
}
int main() {
int sum=0;
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++) {
scanf("%d",&a[i]);
maxx=max(maxx,a[i]);
sum+=a[i];
}
int left,right,mid;
left=0,right=sum+1;//防止意外情况
while(left+1<right) {
mid=(left+right)>>1;//位运算快
if(check(mid))right=mid;
else left=mid;
}
printf("%d",right);
return 0;
}