F. Equalize the Array(思维+前缀和)

https://codeforces.com/contest/1490/problem/F


题意:问最小删除多少个数字使得剩下的数字出现次数都一样。

思路:

开始就是暴力枚举剩下一个次数,这样肯定要用一些东西优化。

预处理一下某个次数有多少个种类的数字出现过。然后枚举的时候就可以O1删除。在枚举的次数后面的数由于次数比他多,不能全删。转化一下,先全删除,再加上多删除的。多删除的预处理一下某个次数有几个种类的数的后缀。

update:以后cf比赛还是不用unordered_map了。改成map就好了

#include<iostream>
#include<vector>
#include<queue>
#include<unordered_map>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+100;
typedef int LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
unordered_map<LL,LL>map1,map2,v;
LL a[maxn],sum[maxn],suf[maxn],p[maxn];
void init(LL n){
   map1.clear();map2.clear();v.clear();
   for(LL i=0;i<n+10;i++) a[i]=sum[i]=suf[i]=p[i]=0;
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  LL t;cin>>t;
  while(t--){
     LL n;cin>>n;
     init(n);
     LL maxv=0;
     for(LL i=1;i<=n;i++){
        cin>>a[i];
        map1[a[i]]++;
        maxv=max(maxv,map1[a[i]]);
     }
     for(LL i=1;i<=n;i++){
        if(map2.count(a[i])) continue;
        map2[a[i]]=1;
        v[map1[a[i]]]++;///统计出现某次数一共有多少数字

     }
     for(LL i=1;i<=maxv;i++) sum[i]=sum[i-1]+v[i]*i;///累加数字总数
     for(LL i=maxv;i>=1;i--) suf[i]=suf[i+1]+v[i]*i;///后缀数字总数
     for(LL i=maxv;i>=1;i--) p[i]=p[i+1]+v[i];///后缀数字种类和
     LL ans=0x3f3f3f3f;
     for(LL i=1;i<=maxv;i++){
        ans=min(ans,sum[i-1]+(suf[i+1]-p[i+1]*i));///前面数字都归0,后面数字先归0,再加上变成和当前次数一样的数目
     }
     cout<<ans<<"\n";
  }
return 0;
}

 

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页