bzoj 4066: 简单题 带部分重构的KDtree

xiaoxiao2021-02-28  5

题意

你有一个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; }
转载请注明原文地址: https://www.6miu.com/read-1400126.html

最新回复(0)