codeforces 1213D2 Equalizing by Division (hard version) 贪心+暴力

こ雲淡風輕ζ 提交于 2019-11-30 00:28:58

http://codeforces.com/problemset/problem/1213/D2
题目大意:给nn个数,每一次操作可以任意选择一个数aia_{i},使ai=ai/2a_{i}=a_{i}/2(向下取整)。问至少需要多少次操作才能使得序列中有kk个相等的数。
思路:排序后暴力计算贡献就行了。设vis[i]vis[i]表示有多少个等于ii的数,设ans[i]ans[i]表示得到vis[i]vis[i]ii最少需要多少次操作,那么对于aia_{i},一直做除2操作,同时更新vis[i]vis[i]ans[i]ans[i],最后遍历visvis序列找到满足vis[i]>=kvis[i]>=k的最小的ans[i]ans[i]即可。注意一定要先对序列从小到大排序。(贪心思想)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define pr pair<int,int>
using namespace std;
typedef long long ll;

const int maxn=2e5+5;

int n,k;
int a[maxn];
int vis[maxn];
int ans[maxn];

int main()
{
    scanf("%d %d",&n,&k);
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        if(++vis[a[i]]>=k)
            vis[0]=1;
    }
    if(vis[0])
    {
        printf("0\n");
        return 0;
    }
    memset(vis,0,sizeof(vis));
    sort(a,a+n);
    int tmp=0;
    for(int i=0;i<n;i++)
    {
        tmp=0;
        while(a[i]>=1&&vis[a[i]]<k)
            ++vis[a[i]],ans[a[i]]+=tmp,++tmp,a[i]>>=1;
    }
    int re=1e9;
    for(int i=1;i<=2e5;i++)
        if(vis[i]==k)
            re=min(re,ans[i]);
    printf("%d\n",re);
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!