看到題完全沒想到是二分(太弱,太渣了),別人告訴本弱這道題是二分依舊不會做。
思路:二分答案ans,然後枚舉j,然後再二分i,求出矩陣中小於等於ans元素的個數……
注意二分時如果l或r出現負數,最好直接向下取整,否則如(-1+0)/2=0,本來是想讓他等於-1的,這樣就可能進入死循環。
#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
#include <stack>
#include <vector>
#include <string.h>
#include <math.h>
#include <queue>
#define msc(X) memset(X,-1,sizeof(X))
#define ms(X) memset(X,0,sizeof(X))
typedef long long LL;
using namespace std;
const LL inf=1ll<<33;
#define f(X,Y) ((X)*(X)+100000*(X)+(Y)*(Y)-100000*(Y)+(X)*(Y))
const double eps=1e-5;
int n;
LL m;
bool judge(LL num)
{
LL cnt=0;
for(int j=1;j<=n;j++)
{
int l=1,r=n+1,mid;
while(l<r){
mid=(l+r)>>1;
if(f((LL)mid,(LL)j)>num) r=mid;
else if(f((LL)mid,(LL)j)<=num&&f((LL)(mid+1),(LL)j)>num)
break;
else l=mid+1;
}
if(r!=mid)
cnt+=mid;//當前<=num有多少個
}
return cnt<m;
}
int main(int argc, char const *argv[])
{
int t;
scanf("%d",&t);
while(t--){
scanf("%d %lld",&n,&m);
LL l=-inf,r=inf;
while(l<r){
LL mid=(LL)(floor((l+r)/2.0+eps));
if(judge(mid)) l=mid+1;
else r=mid;
}
printf("%lld\n",l );
}
return 0;
}