Yandex.Algorithm 2011 Round 2 D - Powerful array (莫队算法)

题意:

给你n个数,进行m个查询,每个查询有一个区间,如果1出现了cnt[1]次,2出现了cnt[2]次,3出现了cnt[3]次…..i出现cnt[i]次,那么就要求输出 1cnt[1]2+2cnt[2]2+3cnt[3]2+...+icnt[i]2

思路:

莫队裸题,用cnt记录下某个数字出现,这样就能o(1)查询了(因为这个题时间要求高,map,vector会t),注意下long long,其他的就是莫队的正常操作了

错误及反思:

在没加inline以前,跑了4800ms+,加了以后3900ms,莫队因为每次转移区间都要调用函数,所以这样的优化极为明显

代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn=200010;
int arr[maxn],pos[maxn],cnt[1000010];
int n,m;
long long now=0,ans[maxn];

struct query{
    int l,r,id;
}Q[maxn];
bool cmp(query a,query b)
{
    if(pos[a.l]==pos[b.l])
        return a.r<b.r;
    return a.l<b.l;
}
inline void update(int num,int p)
{
    now-=1ll*cnt[num]*cnt[num]*num;
    cnt[num]+=p;
    now+=1ll*cnt[num]*cnt[num]*num;
}
int main(){
    scanf("%d%d",&n,&m);
    int block=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&arr[i]);
        pos[i]=i/block;
    }

    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&Q[i].l,&Q[i].r);
        Q[i].id=i;
    }

    sort(Q,Q+m,cmp);

    for(int i=0,l=1,r=0;i<m;i++)
    {
        for(;l<Q[i].l;l++)
            update(arr[l],-1);
        for(;l>Q[i].l;l--)
            update(arr[l-1],1);
        for(;r<Q[i].r;r++)
            update(arr[r+1],1);
        for(;r>Q[i].r;r--)
            update(arr[r],-1);
        ans[Q[i].id]=now;
    }
    for(int i=0;i<m;i++)
        printf("%I64d\n",ans[i]);
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值