CodeForces - 1213D2 Equalizing by Division (hard version)(思维+vector容器 )

题目链接https://vjudge.net/contest/344823#problem/F

在这里插入图片描述
翻译
输入两个数n和k。
每个数可以进行除2的操作。例如:3/2=1,4/2=2
最少进行多少次操作可以保证n个数中最少有k个相同的数。

分析:
利用vector数组,对于每个数,vector存的是得到这个数进行的操作次数。同时记录这个数出现的次数。
对于出现的次数大于k的数,将次数从小到大排序,记录前k个数的和,取其最小值

代码

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int N=2*1e5+10;
int n,k;
int a[N],book[N],mx;
vector<int>v[N];
int main()
{
    scanf("%d%d",&n,&k);
    mx=0;
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        mx=max(mx,a[i]);
        int x=a[i],ans=0;
        while(x)
        {
            book[x]++;
            v[x].push_back(ans);/*x是经过几次变换得到的*/
            x/=2;
            ans++;
        }
    }
    int mi=0x3f3f3f3f;
    for(int i=1; i<=mx; i++)
    {
        if(book[i]>=k)
        {
            int sum=0;
            sort(v[i].begin(),v[i].end());/*从小到大排序,记录前k个最小的次数*/
            for(int j=0; j<k; j++)
                sum+=v[i][j];
            mi=min(mi,sum);
        }
    }
    printf("%d\n",mi);
    return 0;
}
发布了165 篇原创文章 · 获赞 6 · 访问量 5073

猜你喜欢

转载自blog.csdn.net/lylzsx20172018/article/details/103243528
今日推荐