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();
}