题意
你有一个N*N的棋盘,每个格子内有一个整数,初始时的时候全部为0,现在需要维护两种操作:
1 x y A将格子x,y里的数字加上A
2 x1 y1 x2 y2输出x1 y1 x2 y2这个矩形内的数字和
3 终止程序
n<=500000,操作数<=200000,强制在线,空间限制为20M。
分析
如果没有空间限制和强制在线的话,就可以直接离线cdq分治或二维线段树。现在要求强制在线,就只能用KDtree。由于还要支持插入,所以就要像替罪羊树一样暴力重构来维护深度。
剩下的就是模板了。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=200005;
int n,sz,rt,DD,top,stack[N],R;
struct tree{int d[2],mn[2],mx[2],s,size,dir,fa,l,r,val;}t[N];
int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void upd(int x,int y)
{
t[x].mn[0]=min(t[x].mn[0],t[y].mn[0]);
t[x].mx[0]=max(t[x].mx[0],t[y].mx[0]);
t[x].mn[1]=min(t[x].mn[1],t[y].mn[1]);
t[x].mx[1]=max(t[x].mx[1],t[y].mx[1]);
}
void updata(int d)
{
t[d].s=t[t[d].l].s+t[t[d].r].s+t[d].val;
t[d].size=t[t[d].l].size+t[t[d].r].size+1;
t[d].mn[0]=t[d].mx[0]=t[d].d[0];
t[d].mn[1]=t[d].mx[1]=t[d].d[1];
if (t[d].l) upd(d,t[d].l);
if (t[d].r) upd(d,t[d].r);
}
void dfs(int x)
{
if (t[x].l) dfs(t[x].l);
stack[++top]=x;
if (t[x].r) dfs(t[x].r);
}
bool cmp(int x,int y)
{
return t[x].d[DD]<t[y].d[DD]||t[x].d[DD]==t[y].d[DD]&&t[x].d[DD^1]<t[y].d[DD^1];
}
void build(int &d,int l,int r,int D)
{
if (l>r) {d=0;return;}
int mid=(l+r)/2;
DD=D;nth_element(stack+l,stack+mid,stack+r+1,cmp);
d=stack[mid];t[d].dir=D;
build(t[d].l,l,mid-1,D^1);if (t[d].l) t[t[d].l].fa=d;
build(t[d].r,mid+1,r,D^1);if (t[d].r) t[t[d].r].fa=d;
updata(d);
}
void rebuild(int d)
{
int tmp=t[d].fa,op=(d==t[tmp].l?0:1);
top=0;dfs(d);build(d,1,top,t[d].dir);
if (!tmp) rt=d,t[d].fa=0;
else if (!op) t[tmp].l=d;
else t[tmp].r=d;
}
void ins(int &d,int x,int y,int z,int D)
{
if (!d)
{
d=++sz;t[d].dir=D;
t[d].mn[0]=t[d].mx[0]=t[d].d[0]=x;
t[d].mn[1]=t[d].mx[1]=t[d].d[1]=y;
t[d].val=t[d].s=z;t[d].size=1;
return;
}
if (t[d].d[0]==x&&t[d].d[1]==y) {t[d].val+=z;t[d].s+=z;return;}
if (!D)
if (x<t[d].d[0]||x==t[d].d[0]&&y<t[d].d[1]) ins(t[d].l,x,y,z,D^1),t[t[d].l].fa=d;
else ins(t[d].r,x,y,z,D^1),t[t[d].r].fa=d;
else
if (y<t[d].d[1]||y==t[d].d[1]&&x<t[d].d[0]) ins(t[d].l,x,y,z,D^1),t[t[d].l].fa=d;
else ins(t[d].r,x,y,z,D^1),t[t[d].r].fa=d;
updata(d);
if (t[d].size*0.75<max(t[t[d].l].size,t[t[d].r].size)) R=d;
}
bool in(int x1,int x2,int y1,int y2,int x3,int x4,int y3,int y4)
{
return x1>=x3&&x2<=x4&&y1>=y3&&y2<=y4;
}
bool out(int x1,int x2,int y1,int y2,int x3,int x4,int y3,int y4)
{
return x2<x3||x1>x4||y2<y3||y1>y4;
}
int query(int d,int x1,int x2,int y1,int y2)
{
if (!d||out(x1,x2,y1,y2,t[d].mn[0],t[d].mx[0],t[d].mn[1],t[d].mx[1])) return 0;
if (in(t[d].mn[0],t[d].mx[0],t[d].mn[1],t[d].mx[1],x1,x2,y1,y2)) return t[d].s;
int ans=0;
if (t[d].d[0]>=x1&&t[d].d[0]<=x2&&t[d].d[1]>=y1&&t[d].d[1]<=y2) ans+=t[d].val;
return ans+query(t[d].l,x1,x2,y1,y2)+query(t[d].r,x1,x2,y1,y2);
}
int main()
{
n=read();
int ans=0;
while (1)
{
int op=read();
if (op==1)
{
int x=read()^ans,y=read()^ans,z=read()^ans;
ins(rt,x,y,z,0);
if (R) rebuild(R),R=0;
}
else if (op==2)
{
int x1=read()^ans,y1=read()^ans,x2=read()^ans,y2=read()^ans;
printf("%d\n",ans=query(rt,x1,x2,y1,y2));
}
else break;
}
return 0;
}