F. Equalize the Array-(Codeforces Round #702 (Div. 3))-前缀和,后缀和,map的应用

在这里插入图片描述
在这里插入图片描述
题意:就是问你如果让数列里的数字出现的次数都一样,最少需要删除多少个数字
思路:思路我打在了代码注释上,这题不太好想,前后缀和都应用到了。详细的看注释吧
代码:

#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;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值