这题还卡树状数组,一把辛酸泪啊!
考虑维护每个颜色用分块,修改是O(1)查询是sqrt(n)
Code:分块
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define lowbit(x) (x&(-x))
using namespace std;
int sc()
{
int i=0;char c=getchar();
while(c>'9'||c<'0')c=getchar();
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i;
}
struct W{int l,r,a,b,pos;}q[1000001];
int a[100010],bl[100010],cnt[100010];
int block_ans[100100];
int ans[1000010],n,m,block;
bool cmp(W a,W b)
{
return bl[a.l]==bl[b.l]?a.r<b.r:bl[a.l]<bl[b.l];
}
int query(int l,int r)
{
int x=bl[l],y=bl[r],ans=0;
if(x==y)
{
for(int i=l;i<=r;i++)ans+=(bool)cnt[i];
return ans;
}
for(int i=x+1;i<y;i++)ans+=block_ans[i];
for(int i=l;bl[i]==x;i++)ans+=(bool)cnt[i];
for(int i=r;bl[i]==y;i--)ans+=(bool)cnt[i];
return ans;
}
void cal(int x,int f)
{
cnt[x]+=f;
if(f==1&&cnt[x]==1)
block_ans[bl[x]]++;
else if(f==-1&&cnt[x]==0)
block_ans[bl[x]]--;
}
void print(int x)
{
if(x==0){puts("0");return;}
char c[20]; int i=0;
while(x)c[++i]=x%10+'0',x/=10;
while(i)putchar(c[i--]);
puts("");
}
int main()
{
n=sc(),m=sc();block=sqrt(n);
for(int i=1;i<=n;i++)a[i]=sc();
for(int i=1;i<=m;i++)q[i]=(W){sc(),sc(),sc(),sc(),i};
for(int i=1;i<=n;i++)bl[i]=(i-1)/block+1;
sort(q+1,q+m+1,cmp);
int l=q[1].l,r=q[1].l;
cal(a[q[1].l],1);
for(int i=1;i<=m;i++)
{
while(l<q[i].l)cal(a[l++],-1);
while(r>q[i].r)cal(a[r--],-1);
while(l>q[i].l)cal(a[--l],1);
while(r<q[i].r)cal(a[++r],1);
ans[q[i].pos]=query(q[i].a,q[i].b);
}
for(int i=1;i<=m;i++)print(ans[i]);
return 0;
}
树状数组(T掉了)
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<set>
#include<map>
#define lowbit(x) (x&(-x))
using namespace std;
int sc()
{
int i=0;char c=getchar();
while(c>'9'||c<'0')c=getchar();
while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
return i;
}
struct W{int l,r,a,b,pos;}q[1000001];
int a[100010],tr[100010],bl[100010],cnt[100010];
int ans[1000010],n,m,block;
bool cmp(W a,W b)
{
return bl[a.l]==bl[b.l]?a.r<b.r:bl[a.l]<bl[b.l];
}
void change(int x,int f)
{
for(;x<=n;x+=lowbit(x))
tr[x]+=f;
}
int ask(int x)
{
int ans=0;
for(;x;x-=lowbit(x))ans+=tr[x];
return ans;
}
void cal(int x,int f)
{
cnt[x]+=f;
if(f==1&&cnt[x]==1)change(x,1);
if(f==-1&&cnt[x]==0)change(x,-1);
}
void print(int x)
{
if(x==0){puts("0");return;}
char c[20]; int i=0;
while(x)c[++i]=x%10+'0',x/=10;
while(i)putchar(c[i--]);
puts("");
}
int main()
{
n=sc(),m=sc();block=sqrt(n);
for(int i=1;i<=n;i++)a[i]=sc();
for(int i=1;i<=m;i++)q[i]=(W){sc(),sc(),sc(),sc(),i};
for(int i=1;i<=n;i++)bl[i]=(i-1)/block+1;
sort(q+1,q+m+1,cmp);
int l=q[1].l,r=q[1].l;
cal(a[q[1].l],1);
for(int i=1;i<=m;i++)
{
while(l<q[i].l)cal(a[l++],-1);
while(r>q[i].r)cal(a[r--],-1);
while(l>q[i].l)cal(a[--l],1);
while(r<q[i].r)cal(a[++r],1);
ans[q[i].pos]=ask(q[i].b)-ask(q[i].a-1);
}
for(int i=1;i<=m;i++)print(ans[i]);
return 0;
}