【对询问分块】【主席树】bzoj2683 简单题

时间:2022-06-01 23:57:55

对操作序列分块,每S次暴力重建主席树。

当S=sqrt(n*log(n))时,复杂度为O(m*sqrt(n*log(n)))。

在线的。

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
#define N 500001
#define M 200001
struct Point{int x,y,z;};
bool operator < (const Point &a,const Point &b){return a.x<b.x;}
Point t[M];
int en;
struct Node
{
int v,lc,rc;
}T[N*20];
int e,root[M],bel[N],re;
int m,n;
int op[M],xa[M],ya[M],xb[M],yb[M],val[M];
void Insert(int pre,int cur,int p,int v,int l,int r)
{
if(l==r)
{
T[cur].v=T[pre].v+v;
return;
}
int m=(l+r>>1);
if(p<=m)
{
T[cur].lc=++e;
T[cur].rc=T[pre].rc;
Insert(T[pre].lc,T[cur].lc,p,v,l,m);
}
else
{
T[cur].rc=++e;
T[cur].lc=T[pre].lc;
Insert(T[pre].rc,T[cur].rc,p,v,m+1,r);
}
T[cur].v=T[T[cur].lc].v+T[T[cur].rc].v;
}
int Query(int pre,int cur,int ql,int qr,int l,int r)
{
if(ql<=l&&r<=qr) return T[cur].v-T[pre].v;
int m=(l+r>>1),res=0;
if(ql<=m) res+=Query(T[pre].lc,T[cur].lc,ql,qr,l,m);
if(m<qr) res+=Query(T[pre].rc,T[cur].rc,ql,qr,m+1,r);
return res;
}
int main()
{
scanf("%d",&n);
for(int i=1;;++i)
{
scanf("%d",&op[i]);
if(op[i]==1) scanf("%d%d%d",&xa[i],&ya[i],&val[i]);
else if(op[i]==2)
{
scanf("%d%d%d%d",&xa[i],&ya[i],&xb[i],&yb[i]);
if(xa[i]>xb[i]) swap(xa[i],xb[i]);
if(ya[i]>yb[i]) swap(ya[i],yb[i]);
} else break;
}
int sz=(int)sqrt((double)n*log2((double)n));
for(int i=1;;++i)
{
if(op[i]==3) break;
else if(op[i]==2)//query
{
int ans=Query(bel[xa[i]-1],bel[xb[i]],ya[i],yb[i],1,n);
for(int j=(i%sz==0||sz==1)?i-sz+1:i/sz*sz+1;j<=i;++j)
if(op[j]==1&&xa[j]>=xa[i]&&xa[j]<=xb[i]&&ya[j]>=ya[i]&&ya[j]<=yb[i])
ans+=val[j];
printf("%d\n",ans);
}
if(i%sz==0||sz==1)//Rebuild
{
for(int j=1;j<=e;++j) T[j].v=0;
e=en=re=0;
for(int j=1;j<=i;++j)
if(op[j]==1)
t[++en]=(Point){xa[j],ya[j],val[j]};
sort(t+1,t+en+1);
for(int j=1;j<t[1].x;++j) bel[j]=0;
for(int j=1;j<=en;++j)
{
++re;
root[re]=++e;
if(j==en||t[j].x!=t[j+1].x)
{
bel[t[j].x]=e;
int End=(j==en)?n:t[j+1].x-1;
for(int k=t[j].x+1;k<=End;++k)
bel[k]=e;
}
Insert(root[re-1],root[re],t[j].y,t[j].z,1,n);
}
}
}
return 0;
}