二分check法

昨天实验室的一个小比赛的一道题目。你能否体会那种,看着大家都会做,只有自己做不出来的绝望。

上下有界且单调,最大值求最小,最小值求最大,一般用二分搜索可以求解。(我刚知道的…)
模板:

//check()函数
int check(int k){
    
    
	int sum=0,ans=1;
	for(int i=0;i<n;i++){
    
    
		if(sum+a[i]<=k)
			sum+=a[i];
		else {
    
    
			sum=a[i];
			ans++;
		}
	}
	return ans;
}
//二分的结构
while(left<=right){
    
    
	mid=(left+right)/2;
	if(条件)	//用check()函数的返回值进行判断 
		letf=mid+1;
	else
		right=mid-1; 
} 
cout<<left<<endl;//输出结果 

先看例题:


题目描述

对于给定的一个长度为N的正整数数列A−i,现要将其分成M(M≤N)段,并要求每段连续,且每段和的最大值最小。

关于最大值最小:

例如一数列4 2 4 5 1要分成3段

将其如下分段:

[42][45][1]

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

将其如下分段:

[4][24][51]

扫描二维码关注公众号,回复: 13124699 查看本文章

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

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

所以可以得到要将数列42451要分成3段,每段和的最大值最小为6。

输入格式

第1行包含两个正整数N,M。

第2行包含N个空格隔开的非负整数Ai​,含义如题目所述。

输出格式

一个正整数,即每段和最大值最小为多少。

输入样例

5 3
4 2 4 5 1

输出样例

6

说明

对于20%的数据,有N≤10;

对于40%的数据,有N≤1000;

对于100%的数据,有N≤100000,M≤N,Ai​之和不超过10^9。


要求每段和最大值最小,left取这个数列中元素最大的值,right取整个数列的和。那么要求的每段和的最大值的最小值就一定在left和right之间。可以再用二分法进行求解。
这种题目的关键在于理解好check()函数的意义。
先上代码:

#include<iostream>
#include<algorithm>
using namespace std;
int n,m;
int a[100005];

int check(int k){
    
    
	int sum=0,ans=1;//此处ans应当赋值1 
	for(int i=0;i<n;i++){
    
    
		if(sum+a[i]<=k)sum+=a[i];
		else{
    
    
			sum=a[i];
			ans++;
		}
	}
	return ans;
}
int main()
{
    
    
	int left=0,right=0,mid;
	cin>>n>>m;
	for(int i=0;i<n;i++){
    
    
		scanf("%d",&a[i]);
		right+=a[i];
		left=max(left,a[i]);
	}
	while(left<=right){
    
     	//二分法的基本结构
		mid=(left+right)/2;
		if(check(mid)>m)
			left=mid+1;
		else 
			right=mid-1;
	}
	cout<<left<<endl;
	return 0;	
 } 

check()函数中的sum记录每一小段的和(每一段的和都不超过k,也就是当前的mid的值),ans记录这种段共有多少个。
如果ans的值比m大,说明此时mid的值比要求得结果小,所以分出的组数要比要求的组数多。否则反之。根绝返回的结果不断更新left和right的值。最终输出left的值。

check函数中ans的初值要取1。因为最后一个分组中的元素一定不会超过k的值,分组的数量也要加上这个分组。

猜你喜欢

转载自blog.csdn.net/weixin_45382645/article/details/109584768