bzoj3809 Gty的二逼妹子序列

来源:互联网 发布:简易描述linux内核 编辑:程序博客网 时间:2024/04/29 11:12

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();}
0 0
原创粉丝点击