【描述】
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]的权值的种类数。(time limit:8s)
【思路】
本来我想用莫队+树状数组水过去,没想到只有20分。这里简单说一下树状数组的思路:维护每个权值个数,当权值种类有增减时修改树状数组,询问即前缀相减。
20分代码:
#include<bits/stdc++.h>
#define re register
#define mp make_pair
using namespace std;
const int N=1e6+5;
inline int red(){
int data=0;bool w=0; char ch=getchar();
while(ch!='-' && (ch<'0' || ch>'9')) ch=getchar();
if(ch=='-') w=1,ch=getchar();
while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
return w?-data:data;
}int n,m,a[N],b,c[N],col[N],ans[N],cnt[N];
inline void Add(int x,const int&v){
while(x<=n)c[x]+=v,x+=x&(-x);
}
inline int query(int x){
int ret=0;while(x)ret+=c[x],x-=x&(-x);
return ret;
}
inline void add(int pos){!cnt[a[pos]]++&&(Add(a[pos],1),1);}
inline void del(int pos){!--cnt[a[pos]]&&(Add(a[pos],-1),1);}
struct node{int l,r,a,b,id;friend inline bool operator<(const node&a,const node&b){return col[a.l]<col[b.l]||(col[a.l]==col[b.l]&&a.r<b.r);}}q[N];
int main(){n=red();m=red();int s=sqrt(n);
for(int re i=1;i<=n;i++)a[i]=red(),col[i]=(i-1)/s+1;
for(int re i=1;i<=m;i++)q[i]=(node){red(),red(),red(),red(),i};
sort(q+1,q+m+1);int l=1,r=0;
for(int re i=1;i<=m;i++){
while(l>q[i].l)add(--l);while(r<q[i].r)add(++r);
while(r>q[i].r)del(r--);while(l<q[i].l)del(l++);
ans[q[i].id]=query(q[i].b)-query(q[i].a-1);
}for(int re i=1;i<=m;i++)cout<<ans[i]<<"\n";
}
为什么会超时呢?树状数组修改log,询问log。而我们有
m
m
m\sqrt{m}
mm次修改,
m
m
m次询问。这复杂度比较爆炸。我们考虑能不能
O
(
1
)
O(1)
O(1)修改?这里就需要一种思想:由于询问次数远小于修改,可以尝试牺牲询问的时间复杂度,提高修改的效率。这里将树状数组换为对权值分块即可。这样即可实现
O
(
1
)
O(1)
O(1)修改,
O
(
n
)
O(\sqrt{n})
O(n)查询。这样就可以平衡修改和询问的时间。
100分代码:
#include<bits/stdc++.h>
#define re register
using namespace std;
const int N=1e5+5,M=1e6+5;
inline int red(){
int data=0;char ch=getchar();
while(ch<'0'||ch>'9')ch=getchar();
while(ch>='0'&&ch<='9')data=(data<<3)+(data<<1)+(ch^48),ch=getchar();
return data;
}int n,m,a[N],b,c[N>>3|1],col[N],ans[M],cnt[N],s,l=1,r=0;
inline int query(int l,int r){int ret=0;
for(int re i=l,suf=min(r,col[l]*s);i<=suf;++i)ret+=(cnt[i]!=0);
for(int re i=col[l]+1;i<col[r];++i)ret+=c[i];
if(col[l]!=col[r])for(int re i=(col[r]-1)*s+1;i<=r;++i)ret+=(cnt[i]!=0);
return ret;
}void print(int x){if(x>9)print(x/10);putchar(x%10^48);}
inline void add(const int&x){!cnt[x]++&&++c[col[x]];}
inline void del(const int&x){!--cnt[x]&&--c[col[x]];}
struct node{int l,r,a,b,id;friend inline bool operator<(const node&a,const node&b){return col[a.l]<col[b.l]||(col[a.l]==col[b.l]&&a.r<b.r);}}q[M];
int main(){n=red();m=red();s=sqrt(n);
for(int re i=1;i<=n;i++)a[i]=red(),col[i]=(i-1)/s+1;
for(int re i=1;i<=m;i++)q[i]=(node){red(),red(),red(),red(),i};
sort(q+1,q+m+1);
for(int re i=1;i<=m;i++){
while(l>q[i].l)add(a[--l]);while(r<q[i].r)add(a[++r]);
while(r>q[i].r)del(a[r--]);while(l<q[i].l)del(a[l++]);
ans[q[i].id]=query(q[i].a,q[i].b);
}for(int re i=1;i<=m;i++)print(ans[i]),putchar('\n');
}