蓝桥杯2023年-平均（小根堆）

题目描述

有一个长度为 n 的数组（n 是 10 的倍数），每个数 ai 都是区间 [0, 9] 中的整数。小明发现数组里每种数出现的次数不太平均，而更改第 i 个数的代价为bi，他想更改若干个数的值使得这 10 种数出现的次数相等（都等于n/10），请问代价和最少为多少。

思路

平均后每个数的数量ned是确定的：n/10，

我们可以把0~9数字可以分成两类：数量大于ned的数字，数量小于ned的数字。（cnt[i]记录数字的出现次数）

我们的目标就是将 数量大于ned的数字 转化为 数量小于ned的数字。

转换时要使代价最小，我们可以将代价放入小根堆中，依次取 代价最小 且 数量大于ned 的数字 转化为 任意 数量小于ned的数字即可。

代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define pii pair<long long,long long>
int cnt[10];
signed main(){
    int n;cin>>n;
    priority_queue<pii,vector<pii>,greater<pii>>q;
    for(int i=0;i<n;i++){
        int a,b;cin>>a>>b;
        cnt[a]++;
        q.push({b,a});
    }
    int ned=n/10;
    int ans=0;
    for(int i=0;i<10;i++){
        while(cnt[i]<ned){
            while(q.size()&&cnt[q.top().second]<=ned){
                q.pop();
            }
            if(q.size()&&cnt[q.top().second]>ned){
                cnt[i]++;
                cnt[q.top().second]--;
                ans+=q.top().first;
                q.pop();
            }
        }
    }
    cout<<ans;
}