Codeforces 1328 F Make k Equal —— 贪心

This way

题意:

给你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;
}

发布了584 篇原创文章 · 获赞 33 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/tianyizhicheng/article/details/105175510