题目
两次二分
首先对ans进行二分,在([-10^{18},10^{18}])之间
考虑怎么check
对于每个ans,枚举每个(a_i),二分查找有几个(a_j),使得(a_i imes a_j<ans)
对于每对合法的((a_i,a_j)),枚举到(a_i)和(a_j)的时候都被找了一遍,所以最后应该除以2来去重
因为不能出现(a_i=a_j),所以返回的时候要把(i=j)的情况去掉
两个数组a,b
将输入的数在a中按升序排序,在b中按降序排
在第二次二分时,当(a_i<0),用b数组二分(负负得正,所以此时绝对值更大的负时是“大”的,因为它与(a_i)的乘积更大),当(a_igeq 0),用a数组,这样保证了可二分性
通过第一层二分,我们找到的其实是在n个(a_i)两两相乘得到的数中,比从小到大第k个数大的最小数,所以要输出(ans-1)
复杂度(O(nlog nlog 10^{18}))
code.
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<iomanip>
#include<cstring>
#define R register
#define EN std::puts("")
#define LL long long
inline LL read(){
LL x=0,y=1;
char c=std::getchar();
while(c<'0'||c>'9'){if(c=='-') y=0;c=std::getchar();}
while(c>='0'&&c<='9'){x=x*10+(c^48);c=std::getchar();}
return y?x:-x;
}
//a:abs大的负数->abs小的负数->小正数->大正数
//b:大正数->小正数->abs小的负数->abs大的负数
int n,fir;//a[fir]是a数组中第一个非负数
LL k;
LL a[200006],b[200006];
inline int finda(int i,LL ans){
R int l=1,r=n,mid,ret=0;
while(l<=r){
mid=(l+r)>>1;
if(a[mid]*a[i]<ans) ret=mid,l=mid+1;
else r=mid-1;
}
return ret>=i?(ret-1):ret;//如果ret>=i,说明其中有一对是两个a[i]相乘得到,此时要减1
}
inline int findb(int i,LL ans){
R int l=1,r=n,mid,ret=0;
while(l<=r){
mid=(l+r)>>1;
if(b[mid]*a[i]<ans) ret=mid,l=mid+1;
else r=mid-1;
}
return (ret+i>n)?(ret-1):ret;//同理,但b是倒序
}
inline int check(LL ans){
LL nowk=0;
for(R int i=1;i<fir;i++)//负数
nowk+=findb(i,ans);
for(R int i=fir;i<=n;i++)//正数
nowk+=finda(i,ans);
nowk/=2;//去重
return nowk>=k;
}
inline int cmp(int x,int y){
return x<y;
}
int main(){
n=read();k=read();
for(R int i=1;i<=n;i++) a[i]=read();
std::sort(a+1,a+1+n,cmp);
fir=n+1;
for(R int i=1;i<=n;i++)
if(a[i]>=0){fir=i;break;}
for(R int i=1;i<=n;i++) b[n-i+1]=a[i];//存入b数组,对负数二分时使用
R LL l=-1e18,r=1e18,mid,ans;
while(l<=r){
mid=(l+r)/2;
if(check(mid)) ans=mid,r=mid-1;
else l=mid+1;
}
std::printf("%lld",ans-1);
return 0;
}