[二分] 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;
}