bzoj3809: Gty的二逼妹子序列分块+莫队

最新推荐文章于 2019-08-22 21:43:48 发布

lvzelong2014

最新推荐文章于 2019-08-22 21:43:48 发布

阅读量181

点赞数

分类专栏： # 分治结构

本文链接：https://blog.csdn.net/lvzelong2014/article/details/83351294

版权

分治结构专栏收录该内容

32 篇文章 0 订阅

订阅专栏

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)$ 过不去。
发现莫队处理好信息之后询问是 $O (l o g n)$
可以再用一个分块平衡一下。
因为分块可以做到 $O (1)$ 修改， $O(\sqrt n)$ 询问。
莫队的时候对权值分块开桶扫即可。
复杂度 $O(n(\sqrt m +\sqrt n))$

代码

大爱结构体

#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;
}