题意:就是问你如果让数列里的数字出现的次数都一样,最少需要删除多少个数字
思路:思路我打在了代码注释上,这题不太好想,前后缀和都应用到了。详细的看注释吧
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
ll a[N],sum[N],pre[N],q[N];
void init()
{
//memset(a,0,sizeof a);
memset(sum,0,sizeof sum);
memset(pre,0,sizeof pre);
memset(q,0,sizeof q);
//memset(vis,0,sizeof vis);
}
int main()
{
//ios::sync_with_stdio(false);
//cin.tie(0);
//cout.tie(0);
ll t;scanf("%lld",&t);
while(t--)
{
map<ll,ll>m,p,vis;
init();
ll n;scanf("%lld",&n);
ll maxx=0;
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
m[a[i]]++;//每个数字的个数
maxx=max(maxx,m[a[i]]);
}
/*
1 1 2 2 2 2 3 4 4
m[1]=2;m[2]=4;m[3]=1;m[4]=2; maxx=4
*/
for(int i=1;i<=n;i++)
{
if(vis[a[i]]==0)
{
vis[a[i]]++;
p[m[a[i]]]++;
}
}
/*
p[1]=1;p[2]=2;p[4]=1;
*/
sum[0]=0;
for(int i=1;i<=maxx;i++)
{
sum[i]=sum[i-1]+i*p[i];//前缀和
}
/*
sum[1]=1;sum[2]=1+4=5;sum[3]=5+0=5;sum[4]=5+4=9;
*/
pre[maxx+1]=0;
for(int i=maxx;i>=1;i--)
{
pre[i]=pre[i+1]+i*p[i];//后缀和
}
/*
pre[4]=4;pre[3]=4+0=4;pre[2]=4+4=8;pre[1]=8+1=9;
*/
q[maxx+1]=0;
for(int i=maxx;i>=1;i--)
{
q[i]=q[i+1]+p[i];//个数的后缀和
}
/*
q[4]=1;q[3]=1+0=1;q[2]=1+2=3;q[3]=3+1=4;
*/
ll ans=0x3f3f3f3f;
//1 1 2 2 2 2 3 4 4 p[1]=1;p[2]=2;p[4]=1;
for(int i=1;i<=maxx;i++)
{
ans=min(ans,sum[i-1]+pre[i+1]-q[i+1]*i);
/*
0+(8-3*1)=0+5=5;//个数大于等于0的个数+(个数大于等于2在数列里有8个数-个数大于等于2的在数列里一共有3种*1(就是如果是让答案变成1的话,留下的个数))//相减就是需要删除的个数
1+(4-1*2)=1+2=3;//个数大于等于1的个数,前缀和就是无法构成,需要全部删除,后缀和就是数列里所有大于它的数字个数-答案数*种类数,就是可以留下来的,剩下的就是删除的,然后求最小值
5+(4-1*4)=5-0=5;
5+(0-0)=5;
*/
}
printf("%lld\n",ans);
}
return 0;
}