洛谷P2709.小B的詢問 （普通莫隊）

2020.5.18
還是蠻有收穫的，誰也沒想到第一次自己寫莫隊能不看題解1A，激動。

這道題就是維護區間內所有數字出現次數的平方和。這要在平時肯定很難搞，至少我是想不出任何能夠處理這個區間問題的樹狀結構，現在有莫隊算法就可以做了。

首先老一套，分塊，排序，這些都不用講。主要是add和del函數，我們觀察到，k範圍比較小，可以開數組記錄每個數字的出現次數，然後當區間加入一個數字的時候，首先要把上一層的貢獻撤掉，就是cnt[a[x]]的平方，然後給這個數字的貢獻再加上1，之後再在答案里加上平方就可以了，這些東西可以在O(1)時間內從l和r轉移過來，然後套上莫隊，就ac了，我也不知道爲什麼我的答案會比所有的標準答案大1，然後-1就ac了。
上代碼：

#include <bits/stdc++.h>
using namespace std;
#define limit (1000000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步兩步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%lld\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a ; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\administrator01\\CLionProjects\\untitled24\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快讀
void write(ll x){
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}
int n, m, k;
struct node{
    int l, r, qid, blo;
    bool operator<(const node &rhs)const{
        if(blo ^ rhs.blo)return l < rhs.l;
        return blo & 1 ? r < rhs.r : rhs.r < r;
    }
}query[limit];
int cnt[limit],a[limit];
ll ans[limit];
ll res;
ll sq(int x){
    return x * x;
}
void add(int x){
    res -= sq(cnt[a[x]]);
    res += sq(++cnt[a[x]]);
}
void del(int x){
    res -= sq(cnt[a[x]]);
    res += sq(--cnt[a[x]]);
}
int main(){
#ifdef LOCAL
    FOPEN;
    //freopen("C:\\Users\\administrator01\\CLionProjects\\untitled24\\out.txt", "w", stdout);
#endif
    n = read(), m = read(), k = read();
    rep(i ,1, n)a[i] = read();
    int block = int(sqrt(n >= 3 ? n * (2.0 / 3) : n));//分塊
    rep(i, 1, m){
        query[i].l = read(), query[i].r = read(), query[i].qid = i;
        query[i].blo = query[i].l / block;//分成塊
    }
    sort(query + 1, query + 1 + m);
    int l = 0 , r = 0;
    res = 0;
    rep(i ,1, m){
        while(l < query[i].l)del(l++);//縮進
        while(l > query[i].l)add(--l);
        while(r < query[i].r)add(++r);
        while(r > query[i].r)del(r--);
        ans[query[i].qid] = res;
    }
    rep(i ,1, m)write(ans[i] - 1),puts("");
    return 0;
}