【21暑假ACM集训】R2-C2-B Equalizing by Division (hard version) CodeForces - 1213D2

题目链接

考查

贪心

题意

给出 n 个正整数 a1, a2, … , an (1 ≤ ai ≤ 2 * 10^5)，且对每个数都可以进行任意次如下操作：n = n / 2(向下取整)，求操作次数的最小值使得这些数中至少有 k 个数相等。 (1 ≤ k ≤ n ≤ 2 * 10^5) 。

思路

对于数a、b、c，若b、c均可通过若干次操作得到a且b < c，易得操作次数t(b) < t(c)，可知对所有可以变成x的ai，取最小的k个ai时操作总数最小。
故先将输入数据从小到大排序(直接用multiset)，再逐个对ai迭代操作至ai == 1。
用obtain[x]记录已得到的x的个数，time[x]记录得到obtain[x]个x需要的总操作数，对ai在经过x次操作过程后得到的ax，若obtain[ax] < k，则time[ax] += x且obtain[ax]++。
最后从1到max(ai)查找time[x]的最小值，且保证能得到k个x（即obtain[x] == k）。
该方法时间复杂度为O(nlog2n)，且max(n) == 2 * 10^5，故不会超时。

提示

题面最后一句：

Don’t forget that it is possible to have ai=0 after some operations, thus the answer always exists.

实际上ai经若干次操作后总会先变成1再变成0，故无需计算time[0]。

代码

#include <iostream>
#include <cstdio>
#include <set>
using namespace std;

const int maxn = 2e5 + 10;

int obtain[maxn], time[maxn];
multiset<int> numset;
multiset<int>::iterator ite;

int main()
{
    int n, k, tmp, cnt, i, mint, maxi;
    scanf("%d%d", &n, &k);
    for (i = 0; i < n; i++)
    {
        scanf("%d", &tmp);
        numset.insert(tmp);
    }
    for (ite = numset.begin(); ite != numset.end(); ite++)
    {
        tmp = *ite;
        cnt = 0;
        while (tmp)
        {
            if (obtain[tmp] < k)
            {
                obtain[tmp]++;
                time[tmp] += cnt;
            }
            cnt++;
            tmp /= 2;
        }
    }
    mint = time[1];
    for (i = 1, maxi = *(--ite); i <= maxi; i++)
    {
        if (obtain[i] >= k && time[i] < mint)
            mint = time[i];
    }
    cout << mint;
    return 0;
}