题意:
给你n个数,你每次可以让最大的数-1或者让最小的数+1,问你最少需要多少步使得有至少k个相同的数。
题解:
好像没什么必要去写这个题解,因为它太水了,怎么获得2400的评分的
首先我们枚举k个相同的数是什么,它一定是其中的某一个数。然后看小于等于这个数的个数是否>=k,还有大于等于这个数的数的个数是否>=k。有一点要注意就是并不用将全部都变成这个数,就像第二个案例一样,只需要5个2即可,多出来的并不用继续操作。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+5;
ll a[N];
int main()
{
int n,k;
scanf("%d%d",&n,&k);
ll sum=0;
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]),sum+=a[i];
sort(a+1,a+1+n);
ll ans=1e18,pre=0;
int p;
for(int i=1;i<=n;i=p){
p=i;
while(a[p]==a[i]&&p<=n)
sum-=a[i],p++;
int ned=max(0,k-(p-i));
if(ned==0)
ans=0;
if(p>k)
ans=min(ans,(ll)(i-1)*a[i]-pre-(i-1-ned));
if(n-i+1>=k)
ans=min(ans,sum-(ll)(n-p+1)*a[i]-(n-p+1-ned));
ans=min(ans,(ll)(i-1)*a[i]-pre+sum-(ll)(n-p+1)*a[i]-(n-(p-i)-ned));
pre+=(ll)a[i]*(p-i);
}
printf("%lld\n",ans);
return 0;
}