bzoj3781: 小B的询问 (莫队)

3781: 小B的询问

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1242  Solved: 848
[Submit][Status][Discuss]

Description

小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。

 

Input

第一行，三个整数N、M、K。

第二行，N个整数，表示小B的序列。

接下来的M行，每行两个整数L、R。

 

Output

M行，每行一个整数，其中第i行的整数表示第i个询问的答案。

 

Sample Input

6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6

Sample Output

6
9
5
2

HINT

 

对于全部的数据，1<=N、M、K<=50000

 

这个数据范围就非常的莫队……也挺分块的

莫队说白了就是一个滑窗在整个序列上滑来滑去，所以莫队的关键之处就在于端点移动对答案的影响

这道题，假设原区间增加一个数进来，也就是左端点左移或右端点右移，假设这个数的值为x

那么答案中的 cnt[x] ^ 2 就该改成 (cnt[x] + 1) ^ 2

差值为 2 * cnt[x] + 1

删去一个数同理

#include <bits/stdc++.h>
using namespace std;

const int MAXN = 50005;
const int MAXE = 400400;
const int INF = 0x3f3f3f3f;

template <typename T> inline void read(T &x) {
    int ch = getchar();
    bool fg = false;
    for (x = 0; !isdigit(ch); ch = getchar()) {
        if (ch == '-') {
            fg = true;
        }
    }
    for (; isdigit(ch); ch = getchar()) {
        x = x * 10 + ch - '0';
    }
    if (fg) {
        x = -x;
    }
}

int n, belong[MAXN], cnt[MAXN], Q, val[MAXN], sz, K;

struct Query {
    int l, r, pos;
    Query(int _l = 0, int _r = 0, int _p = 0) : l(_l), r(_r), pos(_p){}
    bool operator < (const Query &Ques) const {
        if(belong[Ques.l] != belong[l]) return belong[l] < belong[Ques.l];
        if(belong[Ques.l] & 1) return r < Ques.r;
        return r > Ques.r;
    }
}q[MAXN];

#define LL long long
LL tot, ans[MAXN];

inline void modify(int x, int k) {
    if(k > 0) cnt[x]++, tot += (LL)(cnt[x] << 1ll) - 1ll;
    else cnt[x]--, tot -= (LL)(cnt[x] << 1ll) + 1ll;
}

signed main() {
    read(n), read(Q), read(K);
    sz = (int) sqrt(n);
    for(int i = 1; i <= n; i++) belong[i] = i / sz;
    for(int i = 1; i <= n; i++) read(val[i]);
    for(int i = 1; i <= Q; i++) {
        int x, y; read(x), read(y);
        q[i] = Query(x, y, i);
    }
    sort(q + 1, q + Q + 1);
    int l = 1, r = 0;
    for(int i = 1; i <= Q; i++) {
        while(l > q[i].l) modify(val[--l], 1);
        while(r < q[i].r) modify(val[++r], 1);
        while(l < q[i].l) modify(val[l++], -1);
        while(r > q[i].r) modify(val[r--], -1);
        ans[q[i].pos] = tot;
    }
    for(int i = 1; i <= Q; i++) printf("%lld\n", ans[i]);
    return 0;
}