Comet OJ - Contest #3 (A 比赛 加强版）二分答案

考试的时候同届神犇 JZYshurak 出了一个  n=$10^5$ 的数据加强版. 

刚开始没什么思路，但是忽然想到这个可以转成二分判定+暴力枚举的模型. 

二分 ans, 使得大于等于 ans 的值小于 k 个，这样就能保证只需枚举小于 k 个值了. 

code: 

#include <bits/stdc++.h> 
#define ll long long 
#define N 300002   
#define setIO(s) freopen(s".in","r",stdin) , freopen(s".out","w",stdout)   
using namespace std;        
int n,k;       
ll A[N];                 
int check(ll tmp) 
{
    int i,j;
    ll re=0;      
    for(i=n;i>=1;--i) 
    {   
        j=lower_bound(A+1,A+1+n,tmp-A[i])-A;   
        if(j<i) 
        {   
            re+=i-j;   
        }    
    }
    return re<1ll*k;        
}
int main() 
{  
    int i,j;  
    setIO("onevsk");  
    scanf("%d%d",&n,&k);  
    for(i=1;i<=n;++i) scanf("%lld",&A[i]);  
    sort(A+1,A+1+n);                
    ll l=1, r=2000000001, mid,ans=0;  
    while(l<=r) 
    {
        // 大于 mid 有 k 个                   
        mid=(l+r)>>1;   
        if(check(mid)) ans=mid, r=mid-1;   
        else l=mid+1;      
    }
    ll re=0;   
    int pp=0;  
    for(i=n;i>=1;--i) 
    {
        for(j=i-1;j>=1;--j) 
        {
            if(A[i]+A[j]<ans) break;       
            re+=A[i]+A[j];   
            ++pp;   
        }
    }                 
    re+=1ll*(1ll*k-1ll*pp)*(ans-1);      
    printf("%lld\n",re);      
    return 0; 
}