[luoguP2709] 小B的询问（莫队）

传送门

个数　　1　　2　　3　　4　　5

答案　　1　　4　　9　  16　 25

做差　　1　　3　　5　　7　　9

显然增加一个数只需要增加 ton[a[x]] << 1 | 1 即可

　　减去一个数也减去这个

注意先加减再更新 ton

——代码

 #include <cmath>
 #include <cstdio>
 #include <iostream>
 #include <algorithm>

 const int MAXN = ;
 int n, m, S, k, ans = ;
 int a[MAXN], ton[MAXN], anslist[MAXN];
 struct node
 {
     int l, r, id, num;
 }q[MAXN];

 inline int read()
 {
     int x = ;
     char ch = getchar();
     for(; !isdigit(ch); ch = getchar());
     for(; isdigit(ch); ch = getchar()) x = (x << ) + (x << ) + ch - '';
     return x;
 }

 inline bool cmp(node x, node y)
 {
     return x.id ^ y.id ? x.id < y.id : x.r < y.r;
 }

 int main()
 {
     int i, x, y;
     n = read();
     m = read();
     k = read();
     S = sqrt(n);
     for(i = ; i <= n; i++) a[i] = read();
     for(i = ; i <= m; i++) q[i].l = read(), q[i].r = read(), q[i].num = i;
     for(i = ; i <= m; i++) q[i].id = q[i].l / S + ;
     std::sort(q + , q + m + , cmp);
     x = q[].l, y = q[].l;
     ton[a[x]]++;
     for(i = ; i <= m; i++)
     {
         while(x < q[i].l)
         {
             ton[a[x]]--;
             ans -= ton[a[x]] <<  | ;
             x++;
         }
         while(x > q[i].l)
         {
             x--;
             ans += ton[a[x]] <<  | ;
             ton[a[x]]++;
         }
         while(y > q[i].r)
         {
             ton[a[y]]--;
             ans -= ton[a[y]] <<  | ;
             y--;
         }
         while(y < q[i].r)
         {
             y++;
             ans += ton[a[y]] <<  | ;
             ton[a[y]]++;
         }
         anslist[q[i].num] = ans;
     }
     for(i = ; i <= m; i++) printf("%d\n", anslist[i]);
     return ;
 }