洛谷P1182题解 二分理解
对于给定的一个长度为N的正整数数列 A_{1\sim N}A
1∼N
,现要将其分成 MM(M\leq NM≤N)段,并要求每段连续,且每段和的最大值最小。
关于最大值最小:
例如一数列 4\ 2\ 4\ 5\ 14 2 4 5 1 要分成 33 段。
将其如下分段:
[4\ 2][4\ 5][1]
[4 2][4 5][1]
第一段和为 66,第 22 段和为 99,第 33 段和为 11,和最大值为 99。
将其如下分段:
[4][2\ 4][5\ 1]
[4][2 4][5 1]
第一段和为 44,第 22 段和为 66,第 33 段和为 66,和最大值为 66。
并且无论如何分段,最大值不会小于 66。
所以可以得到要将数列 4\ 2\ 4\ 5\ 14 2 4 5 1 要分成 33 段,每段和的最大值最小为 66。
本题AC代码:
#include<bits/stdc++.h>
using namespace std;
int n,m;
int s[100005];
bool check(int total)
{
int sum=0,step=0;
for(int i=1;i<=n;i++)
{
if(sum+s[i]<=total)sum+=s[i];
else {
sum=s[i];
step++;
}
}
return step+1>m;
}
int main()
{
int maxn=0,total=0;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>s[i];
total+=s[i];
maxn=max(maxn,s[i]);
}
int l=maxn,r=total;
int mid;
while(l<=r)
{
mid=(l+r)/2;
if(check(mid))l=mid+1;
else r=mid-1;
}
cout<<l<<endl;
return 0;
}
二分搜索就是两部分:二分+搜索。那么对应的难点就是如何分和怎么搜索。
关于此题:可以这样想,我们要找一个最大值最小。就是在多个分割情况下,找一种分割方式,使其最大值在所有分割方式里是最小的。
所以,如果要运用二分,我们首先要确定一个二分范围,即确定 l 和 r 。问什么,求什么。题目要我们输出最大最小值,我们就设这个值为mid,而这个mid最小一定是数组中最大的元素,而最大值应该是所有的数组元素加到一起的总和。由此,我们就确定了这段代码:
for(int i=1;i<=n;i++)
{
cin>>s[i];
total+=s[i];
maxn=max(maxn,s[i]);
}
int l=maxn,r=total;
什么二分范围后,我们就要确定怎么分的问题了。
其实这道题和上次那个搬石头的题相似,我们一开始都无法确定一个准确值,但我们可以通过不断二分的过程来确定那个最优解。
仔细分析,我们设的mid是最大值,那么我们必须保证每一个段的和都应该小于或等于这个最大值。如图:
如果mid减小,我们的分组可能会增多,如果mid增大,我们的分组可能会相应的减少。那么当我们的mid处于一种玄妙的状态:即增加一点就会使分组增多的时候,这时候mid就处于如下位置:
而本题要我们求最大最小值,但我们找到一个这样的边界点后,我们还要考虑这个边界点是不是最优解。因为我们求的是最小值,所以我们要将mid减少。
综上我们可以写出check函数:
bool check(int total)
{
int sum=0,step=0;
for(int i=1;i<=n;i++)
{
if(sum+s[i]<=total)sum+=s[i];
else {
sum=s[i];
step++;
}
}
return step+1>m;//这里注意,不能遗漏了最后一段,所以要step+1
}