[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 = 50001;
int n, m, S, k, ans = 1;
int a[MAXN], ton[MAXN], anslist[MAXN];
struct node
{
    int l, r, id, num;
}q[MAXN];

inline int read()
{
    int x = 0;
    char ch = getchar();
    for(; !isdigit(ch); ch = getchar());
    for(; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + ch - '0';
    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 = 1; i <= n; i++) a[i] = read();
    for(i = 1; i <= m; i++) q[i].l = read(), q[i].r = read(), q[i].num = i;
    for(i = 1; i <= m; i++) q[i].id = q[i].l / S + 1;
    std::sort(q + 1, q + m + 1, cmp);
    x = q[1].l, y = q[1].l;
    ton[a[x]]++;
    for(i = 1; i <= m; i++)
    {
        while(x < q[i].l)
        {
            ton[a[x]]--;
            ans -= ton[a[x]] << 1 | 1;
            x++;
        }
        while(x > q[i].l)
        {
            x--;
            ans += ton[a[x]] << 1 | 1;
            ton[a[x]]++;
        }
        while(y > q[i].r)
        {
            ton[a[y]]--;
            ans -= ton[a[y]] << 1 | 1;
            y--;
        }
        while(y < q[i].r)
        {
            y++;
            ans += ton[a[y]] << 1 | 1;
            ton[a[y]]++;
        }
        anslist[q[i].num] = ans;
    }
    for(i = 1; i <= m; i++) printf("%d
", anslist[i]);
    return 0;
}