bzoj3236 作业 莫队+树状数组

莫队+树状数组

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#define N 100055
#define M 1000066
using namespace std;
int gy[N],be[N],c[M],cc[M],n,m,nn,maxn,tot,num[N];
struct Query{
int l,r,a,b,id,ans1,ans2;
}qr[M];
bool cmp1(Query a,Query b){
if(be[a.l]==be[b.l])
return a.r<b.r;
return be[a.l]<be[b.l];
}
bool cmp2(Query a,Query b){
return a.id<b.id;
}
int lowbit(int x){
return x&(-x);
}
void add(int x,int y){
int xx=x;
if(y==1&&++num[xx]==1){
while(xx<=maxn){
cc[xx]++;
xx+=lowbit(xx);
}
}
if(y==-1&&--num[xx]==0){
while(xx<=maxn){
cc[xx]--;
xx+=lowbit(xx);
}
}
while(x<=maxn){
c[x]+=y;
x+=lowbit(x);
}
}
int query(int x){
if(x>maxn) x=maxn;
int ans=0;
while(x){
ans+=c[x];
x-=lowbit(x);
}
return ans;
}
int query1(int x){
if(x>maxn) x=maxn;
int ans=0;
while(x){
ans+=cc[x];
x-=lowbit(x);
}
return ans;
}
void work(){
int l=1,r=0;tot=0;
for(int i=1;i<=m;i++){
while(l<qr[i].l) add(gy[l++],-1);
while(l>qr[i].l) add(gy[--l],1);
while(r<qr[i].r) add(gy[++r],1);
while(r>qr[i].r) add(gy[r--],-1);
qr[i].ans1=query(qr[i].b)-query(qr[i].a-1);
qr[i].ans2=query1(qr[i].b)-query1(qr[i].a-1);
}
}
int main()
{
scanf("%d%d",&n,&m); nn=(int)sqrt(n);
for(int i=1;i<=n;i++){
scanf("%d",&gy[i]);
be[i]=(i-1)/nn+1;
maxn=max(maxn,gy[i]);
}
int l,r,a,b;
for(int i=1;i<=m;i++){
scanf("%d%d%d%d",&l,&r,&a,&b);
qr[i].l=l; qr[i].r=r;
qr[i].a=a; qr[i].b=b;
qr[i].id=i;
}
sort(qr+1,qr+m+1,cmp1);
work();
sort(qr+1,qr+m+1,cmp2);
for(int i=1;i<=m;i++)
printf("%d %d\n",qr[i].ans1,qr[i].ans2);
return 0;
}