关于莫队算法推荐一个大佬博客:http://blog.csdn.net/hzj1054689699/article/details/51866615
这是我见过最玄学的一个算法,号称可以解决一切区间问题。主要思想就是将询问分块排序,当前询问结果可以由上一个询问经过较少的变化得到。
影响算法效率的主要因素是block的选取。一般会选择sqrt(n)。据说最佳选取为。
题目链接:CodeForces - 86D
代码:
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;
#define MAXN 200005
int n,m,block;
long long a[MAXN],cnt[1000005];
long long ans,s[MAXN];
struct query
{
int l,r,id;
}q[MAXN];
bool cmp(query x,query y)
{
if(x.l/block!=y.l/block)
return x.l<y.l;
return x.r<y.r;
}
void add(long long x)
{
ans-=cnt[x]*cnt[x]*x;
cnt[x]++;
ans+=cnt[x]*cnt[x]*x;
}
void move(long long x)
{
ans-=cnt[x]*cnt[x]*x;
cnt[x]--;
ans+=cnt[x]*cnt[x]*x;
}
int main()
{
while(~scanf("%d%d",&n,&m))
{
block=sqrt(n)+1000;
memset(cnt,0,sizeof cnt);
for(int i=1;i<=n;i++)
scanf("%lld",&a[i]);
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);
int L=1,R=0;
ans=0;
for(int i=0;i<m;i++)
{
while(L<q[i].l) move(a[L++]);
while(L>q[i].l) add(a[--L]);
while(R<q[i].r) add(a[++R]);
while(R>q[i].r) move(a[R--]);
s[q[i].id]=ans;
}
for(int i=0;i<m;i++)
printf("%lld\n",s[i]);
}
return 0;
}