[二分] atcoder 155D

题目

题目链接：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;
}