【BZOJ3809】Gty的二逼妹子序列 （莫队+分块）

内存：28MB
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。

建议使用输入/输出优化。

题解：
内存太小，什么树套树之类的应该是MLE了，还好我看见了……
首先很容易想到莫队加树状数组，但问题是树状数组是logn的，总复杂度O(m√n×log(n))，应该会TLE。所以我们要用使用少的复杂度换多的。使用分块就可以达到这一效果。分块的修改是O(√n)的，但查询是O(1)的，这题没修改，因此总复杂度是O(m√n),可以过。
代码如下：

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<math.h>
#define ll long long
#define inf 0x7f7f7f7f
#define N 100005
#define M 1000005
using namespace std;
ll read()
{
    ll x=0,f=1;
    char c=getchar();
    while(c<'0' || c>'9') {if(c=='-') f=-1;c=getchar();}
    while(c<='9' && c>='0') {x=x*10+c-'0';c=getchar();}
    return x*f;
}
int n,m,a[N],blo,cnt; 
int ans[M],bl[N],l[1005],r[1005],c[N],bcnt[1005];
struct que
{
    int id,l,r,a,b;
    bool operator <(const que& b) const
    {
        if(bl[l]!=bl[b.l]) return l<b.l;
        return r<b.r;
    }
}q[M];
int query(int x,int y)
{
    int ret=0,ba=bl[x],bb=bl[y];
    for(int i=ba+1;i<bb;i++) ret+=bcnt[i];
    if(ba==bb)
    {
        for(int i=x;i<=y;i++) 
        if(c[i]) ret++;
    }
    else
    {
        for(int i=x;i<=r[bl[x]];i++) if(c[i]) ret++;
        for(int i=l[bl[y]];i<=y;i++) if(c[i]) ret++;
    }
    return ret;
}
void add(int x)
{
    c[x]++;
    if(c[x]==1) bcnt[bl[x]]++;
}
void del(int x)
{
    c[x]--;
    if(!c[x]) bcnt[bl[x]]--;
}
void solve()
{
    int l=1,r=0;
    for(int i=1;i<=m;i++)
    {
        while(l<q[i].l) del(a[l++]);
        while(r<q[i].r) add(a[++r]);
        while(l>q[i].l) add(a[--l]);
        while(r>q[i].r) del(a[r--]);
        ans[q[i].id]=query(q[i].a,q[i].b);
    }
}
int main()
{
    n=read(),m=read();
    blo=sqrt(n/2);
    cnt=n/blo+n%blo!=0;
    for(int i=1;i<=n;i++) bl[i]=(i-1)/blo+1;
    for(int i=1;i<=n;i++)
    {
        r[bl[i]]=i;
        if(!l[bl[i]]) l[bl[i]]=i;
    }
    for(int i=1;i<=n;i++) a[i]=read();
    for(int i=1;i<=m;i++)
    {
        q[i].l=read(),q[i].r=read();
        q[i].a=read(),q[i].b=read(),q[i].id=i;
    }
    sort(q+1,q+m+1);
    solve();
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}

莫队和分块真是好搭档啊，后面就能看出来了。