题目
题目链接:https://atcoder.jp/contests/abc155/tasks/abc155_d
思路
题目大意是数组里面所有数两两相乘,找出第k小
先看数据范围,暴力做必死无疑
整容刀梯我们可以先从相乘得到的值开始二分,然后再二分数组里有多少小于它的值。
代码
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define int long long
#define ll long long
using namespace std;
const int INF = 0x3f3f3f3f;
//int csp=1e-10;
int a[200010];
signed main(){
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
int n,k;
cin>>n>>k;
for(int i=0;i<n;i++) cin>>a[i];
sort(a,a+n);
int l=-1e18,r=1e18;
while(l+1!=r){
//cout<<1<<endl;
int mid=(l+r)/2;
int cnt=0;
for(int i=0;i<n;i++){
if(a[i]<0){
int l1=i,r1=n;
while(l1+1!=r1){
int mmid=(l1+r1)/2;
if(a[i]*a[mmid]<=mid) r1=mmid;
else l1=mmid;
}//这里的二分r是找的有多少个大于mid
cnt+=n-r1;//这里就是有多少个小于的了
//cout<<1<<endl;
}
else{
int l1=i,r1=n;
while(l1+1!=r1){
int mmid=(l1+r1)/2;
if(a[i]*a[mmid]<=mid) l1=mmid;
else r1=mmid;
}//l1是有少小于,但是要减去i之前的
cnt+=l1-i;
}
// cout<<cnt<<endl;
}
// cout<<cnt<<endl;
if(cnt<k) l=mid;
else r=mid;
}
cout<<r<<endl;
return 0;
}