bzoj3809: Gty的二逼妹子序列

Description

Autumn和Bakser又在研究Gty的妹子序列了！但他們遇到了一個難題。
對於一段妹子們，他們想讓你幫忙求出這之內美麗度∈[a,b]的妹子的美麗度的種類數。
為了方便，我們規定妹子們的美麗度全都在[1,n]中。
給定一個長度為n(1<=n<=100000)的正整數序列s(1<=si<=n)，對於m(1<=m<=1000000)次詢問“l,r,a,b”，每次輸出sl…sr中，權值∈[a,b]的權值的種類數。

Input

第一行包括兩個整數n,m(1<=n<=100000,1<=m<=1000000),表示數列s中的元素數和詢問數。
第二行包括n個整數s1…sn(1<=si<=n)。
接下來m行,每行包括4個整數l,r,a,b(1<=l<=r<=n,1<=a<=b<=n),意義見題目描述。
保證涉及的所有數在C++的int內。
保證輸入合法。

Output

對每個詢問，單獨輸出一行，表示sl…sr中權值∈[a,b]的權值的種類數。

Sample Input

10 10
4 4 5 1 4 1 5 1 2 1
5 9 1 2
3 4 7 9
4 4 2 5
2 3 4 7
5 10 4 4
3 9 1 1
1 4 5 9
8 9 3 3
2 2 1 6
8 9 1 4

Sample Output

2
0
0
2
1
1
1
0
1
2

HINT

樣例的部分解釋：
5 9 1 2
子序列為4 1 5 1 2
在[1,2]裡的權值有1,1,2，有2種，因此答案為2。
3 4 7 9
子序列為5 1
在[7,9]裡的權值有5，有1種，因此答案為1。
4 4 2 5
子序列為1
沒有權值在[2,5]中的，因此答案為0。
2 3 4 7
子序列為4 5
權值在[4,7]中的有4,5，因此答案為2。
建議使用輸入/輸出優化。

分析

莫隊+樹狀陣列的思路很顯然。
但是複雜度 $O(n\sqrt{m}logn$

程式碼

大愛結構體

#include<bits/stdc++.h>
const int M = 1e6 + 10, N = 1e5 + 10, Bs = 320;
int ri() {
    char c = getchar(); int x = 0, f = 1; for(;c < '0' || c > '9'; c = getchar()) if(c == '-') f = -1;
    for(;c >= '0' && c <= '9'; c = getchar()) x = (x << 1) + (x << 3) - '0' + c; return x * f;
}
int a[N], id[N], l[Bs], r[Bs], A[M], tot, B;
struct Block {
    int cnt[Bs]; int sum;
    void Ins(int x, int p) {sum -= cnt[x] ? 1 : 0; sum += (cnt[x] += p) ? 1 : 0;}
    int Sum(int l, int r) {
        int x = 0;
        for(int i = l;i <= r; ++i) x += cnt[i] ? 1 : 0;
        return x;
    }
};
struct Ask {int l, r, a, b, id;void in(int i) {l = ri(); r = ri(); a = ri(); b = ri(); id = i;}};
bool cmp(Ask a, Ask c) {
    int ba = a.l / B, bc = c.l / B;
    return ba == bc ? ((ba & 1) ? a.r < c.r : a.r > c.r) : ba < bc;
}
struct MT {
    Block b[Bs]; Ask q[M]; int n;
    void Add(int v, int p) {b[id[v]].Ins(v - l[id[v]], p);}
    int Query(int st, int ed) {
        int x = id[st], y = id[ed];
        if(x == y) return b[x].Sum(st - l[x], ed - l[x]);
        int sum = b[x].Sum(st - l[x], r[x] - l[x]) + b[y].Sum(0, ed - l[y]);
        for(int i = x + 1;i < y; ++i) sum += b[i].sum;
        return sum;
    }
    void Work() {
        std::sort(q + 1, q + n + 1, cmp);
        int L = 1, R = 0;
        for(int i = 1;i <= n; ++i) {
            for(;R < q[i].r;) Add(a[++R], 1);
            for(;L > q[i].l;) Add(a[--L], 1);
            for(;L < q[i].l;) Add(a[L++], -1);
            for(;R > q[i].r;) Add(a[R--], -1);
            A[q[i].id] = Query(q[i].a, q[i].b);
        }
    }
}mt;
int main() {
    int n = ri(); B = sqrt(n); mt.n = ri();
    for(int i = 1;i <= n; ++i) a[i] = ri();
    for(int i = 1;i <= n; ++i) {
        id[i] = (i - 1) / B + 1;
        if(!l[id[i]]) l[id[i]] = i; r[id[i]] = i;
    }
    B = sqrt(mt.n);
    for(int i = 1;i <= mt.n; ++i) mt.q[i].in(i);
    mt.Work();
    for(int i = 1;i <= mt.n; ++i) printf("%d\n", A[i]);
    return 0;
}