bzoj 3781 小B的询问（莫队算法）

【题意】

若干个询问sigma{ cnt[i]^2 } cnt[i]表示i在[l,r]内的出现次数。

【思路】

莫队算法，裸题。

一个cnt数组即可维护插入与删除。

【代码】

 #include<set>
 #include<cmath>
 #include<queue>
 #include<vector>
 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #define trav(u,i) for(int i=front[u];i;i=e[i].nxt)
 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
 using namespace std;

 typedef long long ll;
 const int N = 5e5+;

 ll read()
 {
     char c=getchar();
     ll f=,x=;
     while(!isdigit(c)) {
         if(c=='-') f=-; c=getchar();
     }
     while(isdigit(c))
         x=x*+c-'',c=getchar();
     return x*f;
 }

 struct Node
 {
     int id,pos,l,r;
     bool operator < (const Node& rhs) const
     {
         return pos<rhs.pos || (pos==rhs.pos&&r<rhs.r);
     }
 } q[N];

 int n,m,K,a[N]; ll ans[N],cnt[N],now;

 void upd(int x,int v)
 {
     now-=cnt[a[x]]*cnt[a[x]];
     cnt[a[x]]+=v;
     now+=cnt[a[x]]*cnt[a[x]];
 }

 int main()
 {
 //    freopen("in.in","r",stdin);
 //    freopen("out.out","w",stdout);
     n=read(),m=read(),K=read();
     FOR(i,,n) a[i]=read();
     int B=sqrt(n);
     FOR(i,,m)
     {
         q[i].l=read(),
         q[i].r=read();
         q[i].id=i;
         q[i].pos=(q[i].l-)/B+;
     }
     sort(q+,q+m+);
     int l=,r=;
     FOR(i,,m)
     {
         while(l<q[i].l) upd(l++,-);
         while(l>q[i].l) upd(--l,);
         while(r>q[i].r) upd(r--,-);
         while(r<q[i].r) upd(++r,);
         ans[q[i].id]=now;
     }
     FOR(i,,m) printf("%lld\n",ans[i]);
     return ;
 }