CF1328F-Make k equal （补）(二分)

题目链接：

https://codeforces.com/contest/1328/problem/F

思路：

题目大意是你可以进行两种操作，一种是将最大数减1，另一种是将最小数加1，问你最少经过多少次操作，能使序列中存在k个相同的数

我们先将数组进行排序，然后用前缀和对数组进行预处理。

假设我们以a[i]这个数为标准，通过二分找到第一个大于a[i]的位置index，对于大于a[i]的数来说，我们首先通过操作将

a[index]-a[n]中的所有数变成a[i]+1，那么这一过程中消耗的操作数为(sum[n]-sum[index-1])-(n-index+1)*(x+1)，

若原数组中有num个a[i]这样的数，那么我们接下来要消耗的操作数为min(k-num,n-index+1)，如果此时a[i]的个数仍然

小于K，我们再考虑对小于a[i]的数进行操作，这里不再赘述，由于先找小的可能比先找大的更优，所以我又分情况讨论了，

代码很复杂，但看别人的代码好像很简单

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=2e5+5;
const ll INF=3e18;
ll a[MAXN],n,k,sum[MAXN];
ll ans=INF;
map<int,int>m;
ll check1(int x)
{
    ll res=0;ll index;ll num=m[x];
    index=upper_bound(a+1,a+1+n,x)-a;
    res+=(sum[n]-sum[index-1])-(n-index+1)*(x+1);
    res+=min(k-num,n-index+1),num+=min(k-num,n-index+1);
    if(num<k)
    {
        index=lower_bound(a+1,a+1+n,x)-a-1;//cout<<index<<" "<<num<<" "<<endl;
        res+=index*(x-1)-sum[index];
        res+=k-num;
    }
    return res;
}
ll check2(int x)
{
    ll res=0;ll index;ll num=m[x];
    index=lower_bound(a+1,a+1+n,x)-a-1;
    res+=index*(x-1)-sum[index];
    res+=min(k-num,index),num+=min(k-num,index);
    if(num<k)
    {
        index=upper_bound(a+1,a+1+n,x)-a;//cout<<index<<" "<<num<<" "<<endl;
        res+=(sum[n]-sum[index-1])-(n-index+1)*(x+1);
        res+=k-num;
    }
    return res;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];m[a[i]]++;
        if(m[a[i]]>=k)
        {
            cout<<"0"<<endl;
            return 0;
        }
    }
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++)
        sum[i]=sum[i-1]+a[i];
    for(int i=1;i<=n;i++)
        ans=min(ans,min(check1(a[i]),check2(a[i])));
    cout<<ans<<endl;
    return 0;
}