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]的权值的种类数。

首先可以把询问莫队，然后用树状数组维护，复杂度O(m* sqrt(n)* logn)。
但是注意到很多log的复杂度都被浪费了，因为sqrt(n)次修改才会有一次询问，所以可以考虑平衡规划。
最理想的情况就是O(1)修改O(sqrt(n))查询，这样没有复杂度被浪费。对n个数按权值进行分块即可。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int rd()
{
    int x=0;
    char c=getchar();
    while (c<'0'||c>'9') c=getchar();
    while (c>='0'&&c<='9')
    {
        x=x*10+c-'0';
        c=getchar();
    }
    return x;
}
struct query
{
    int l,r,a,b,num;
}q[1000010];
int a[100010],L[1010],R[1010],ans[1000010],tot[1010],cnt[100010],bel[100010],m,n,T;
bool cmp(query q1,query q2)
{
    return bel[q1.l]==bel[q2.l]?q1.r<q2.r:q1.l<q2.l;
}
void init()
{
    int i,j,l,r;
    n=rd();
    m=rd();
    for (i=1;i<=n;i++)
      a[i]=rd();
    for (i=1;i<=m;i++)
    {
        q[i].l=rd();
        q[i].r=rd();
        q[i].a=rd();
        q[i].b=rd();
        q[i].num=i;
    }
    T=sqrt(n);
    for (i=1;i<=T;i++)
    {
        L[i]=R[i-1]+1;
        R[i]=i==T?n:L[i]+T-1;
        for (j=L[i];j<=R[i];j++)
          bel[j]=i;
    }
    sort(q+1,q+m+1,cmp);
}
void inc(int x)
{
    cnt[x]++;
    if (cnt[x]==1) tot[bel[x]]++;
}
void dec(int x)
{
    cnt[x]--;
    if (cnt[x]==0) tot[bel[x]]--;
}
int qry(int l,int r)
{
    int i,j,ret=0;
    for (i=1;i<=T;i++)
      if (l<=L[i]&&R[i]<=r)
        ret+=tot[i];
      else
        for (j=max(l,L[i]);j<=r&&j<=R[i];j++)
          if (cnt[j])
            ret++;
    return ret;
}
void solve()
{
    int i,j;
    for (i=1;i<=m;i++)
    {
        if (q[i].l>q[i-1].l)
          for (j=q[i-1].l;j<q[i].l;j++)
            dec(a[j]);
        else
          for (j=q[i].l;j<q[i-1].l;j++)
            inc(a[j]);
        if (q[i].r>q[i-1].r)
          for (j=q[i-1].r+1;j<=q[i].r;j++)
            inc(a[j]);
        else
          for (j=q[i].r+1;j<=q[i-1].r;j++)
            dec(a[j]);
        ans[q[i].num]=qry(q[i].a,q[i].b);
    }
    for (i=1;i<=m;i++)
      printf("%d\n",ans[i]);
}
int main()
{
    init();
    solve();
}