(POJ - 3685)Matrix(二分套二分)

 

中文题意：有一个N阶方阵 第i行,j列的值Aij =i^2 + 100000 × i + j^2 - 100000 × j + i × j,需要找出这个方阵的第M小值.

先观察这个等式可以发现，Aij随i的增加而增加，所以我们就可以通过二分x来求取答案，每次二分的内容就是判断比x小的数的数目是否大于等于m，如果数目大于等于m，则当前二分值是大于等于答案的，反之当前二分值是小于答案的，具体二分过程如下：

我们暴力枚举列数，每一列的数是单调递增的，所以对于每一列我们可以二分枚举，这样总的复杂度就是n*log(n)*log(q)（q为答案可能值的范围大小）,细节比较多，下面附上代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
ll n,m;
ll cal(ll i,ll j)
{
	return i*i+100000*i+j*j-100000*j+i*j;
}
bool check(ll x)//若小于等于x的数大于等于m个则返回true，反之返回false 
{
	ll cnt=0;
	for(int i=1;i<=n;i++)//枚举列
	{
		ll l=0,r=n,mid;//二分枚举每一列中小于x的数的数目
		if(l==r) return cal(1,1)<=x;
		while(l<r)
		{
			mid=l+r+1>>1;
			if(cal(mid,i)<=x) l=mid;//找出大于等于x的最后一个数 
			else r=mid-1;
		}
		cnt+=l;
		if(cnt>=m) return true;
	}
	return false;
}
int main()
{
	int T;
	cin>>T;
	while(T--)
	{
		cin>>n>>m;
		ll l=-9999999999,r=9999999999,mid;//枚举数 
		while(l<r)
		{
			mid=l+r>>1;
			if(check(mid)) r=mid;//说明mid偏大
			else l=mid+1;
		}
		printf("%lld\n",l);
	}
	return 0;
}