中文题意:有一个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;
}