当前位置: 代码迷 >> 综合 >> Yandex.Algorithm 2011 Round 2 D - Powerful array (莫队算法)
  详细解决方案

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

热度:25   发布时间:2023-12-17 03:41:45.0

题意:

给你n个数,进行m个查询,每个查询有一个区间,如果1出现了cnt[1]次,2出现了cnt[2]次,3出现了cnt[3]次…..i出现cnt[i]次,那么就要求输出 1?cnt[1]2+2?cnt[2]2+3?cnt[3]2+...+i?cnt[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]);
}
  相关解决方案