洛谷P1631 序列合并 二分做法

洛谷P1631

之前也写的二分一直过不了，以为不行，然后看了这篇laorui的题解后发现只要把数组开大一点就能过。

做了一点优化之后决定写一篇题解 因为他的代码过于清奇

思路就是二分去找数，只要这里面两两求和的对满足有n个就是true，否则就是false，因为单调增的原因只要找n个数所以judge只需要n的复杂度最后复杂度就是nlogm

但是要考虑的是可能二分judge里的退出条件设置小了，可能提前退出了，前面选中的对可能有重复，小的对可能在后面就选不到小的了。

所以得扩大退出条件，大胆假设只需要二倍的n就行了，虽然不知道怎么证明。hoho不过总之是过了。

以及要注意二分之后的目标数组不一定是答案，因为最后在judge里跑的不一定是正确的数。

还有就是二分的初始值，左端点是等于a[0]+b[0],右端点只需要等于a[d]+b[d]就行了，d=ceil(sqrt(n))-1，比这个数大的时候是显然会有超过n对的。

主要看代码

#include<bits/stdc++.h>
using namespace std;
const int max_n=2e5+5;
int a[max_n],b[max_n];
int ans[max_n];
int n,cnt,lim;
bool judge(int max1)
{
	int i,j;
	cnt=0;
	for(i=0;i<n;i++){
		for(j=0;j<n;j++){
			if(a[i]+b[j]<=max1){
				ans[cnt++]=a[i]+b[j];
				if(cnt>=lim) return true;
			}else{
				break;//利用单调性保证复杂度
			}
		}
	}
	if(cnt>=n) return true;
	else return false;
}
int main()
{
	int i,j;
	scanf("%d",&n);
	lim=n<<1;//lim退出条件选两倍大
	for(i=0;i<n;i++){
		scanf("%d",a+i);
	}
	for(j=0;j<n;j++){
		scanf("%d",b+j);
	}
	int d=ceil(sqrt(n))-1;//求右端点的位置
	int l=a[0]+b[0],r=a[d]+b[d],mid;
	int ans1=0;
	while(l<=r){
		mid=l+((r-l)>>1);
		if(judge(mid)){
			ans1=mid;
			r=mid-1;
		}else{
			l=mid+1;
		}
	}
    
	//结束之后要拿ans1再跑一遍，因为最后执行的mid不一定是满足的
	judge(ans1);
	
	sort(ans,ans+cnt);//排序之后输出前n个
	for(i=0;i<n;i++){
		printf("%d ",ans[i]);
	}
	return 0;
}