http://codeforces.com/problemset/problem/1213/D2
题目大意:给
n
n
n个数,每一次操作可以任意选择一个数
a
i
a_{i}
ai,使
a
i
=
a
i
/
2
a_{i}=a_{i}/2
ai=ai/2(向下取整)。问至少需要多少次操作才能使得序列中有
k
k
k个相等的数。
思路:排序后暴力计算贡献就行了。设
v
i
s
[
i
]
vis[i]
vis[i]表示有多少个等于
i
i
i的数,设
a
n
s
[
i
]
ans[i]
ans[i]表示得到
v
i
s
[
i
]
vis[i]
vis[i]个
i
i
i最少需要多少次操作,那么对于
a
i
a_{i}
ai,一直做除2操作,同时更新
v
i
s
[
i
]
vis[i]
vis[i]和
a
n
s
[
i
]
ans[i]
ans[i],最后遍历
v
i
s
vis
vis序列找到满足
v
i
s
[
i
]
>
=
k
vis[i]>=k
vis[i]>=k的最小的
a
n
s
[
i
]
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);
}