题目
k-d树
题解
注意:程序中的D是全局变量,用于排序中的比较
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=200002;
int n,m,x,y,ans,opt,A,xx,yy,D,i;
inline char gc(){
static char buf[100000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
#define gc getchar
inline int read(){
int x=0,fl=1;char ch=gc();
for (;ch<48||ch>57;ch=gc())if(ch=='-')fl=-1;
for (;48<=ch&&ch<=57;ch=gc())x=(x<<3)+(x<<1)+(ch^48);
return x*fl;
}
inline void wri(int a){if(a>=10)wri(a/10);putchar(a%10|48);}
inline void wln(int a){if(a<0)a=-a,putchar('-');wri(a);puts("");}
inline void Min(int &x,int y){if (y<x) x=y;}
inline void Max(int &x,int y){if (y>x) x=y;}
struct P{
int l,r,d[2],mn[2],mx[2],v,sum;
int &operator[](int x){
return d[x];
}
friend bool operator==(P a,P b){
return a.d[0]==b.d[0] && a.d[1]==b.d[1];
}
friend bool operator<(P a,P b){
return a[D]<b[D];
}
}p[N];
bool in(int x1,int y1,int x2,int y2,int X1,int Y1,int X2,int Y2){
return x1<=X1 && X2<=x2 && y1<=Y1 && Y2<=y2;
}
bool out(int x1,int y1,int x2,int y2,int X1,int Y1,int X2,int Y2){
return X2<x1 || X1>x2 || Y2<y1 || Y1>y2;
}
struct node{
P now,t[N];
int rt,cnt;
void update(int k){
int l=t[k].l,r=t[k].r;
for (int i=0;i<2;i++){
t[k].mn[i]=t[k].mx[i]=t[k][i];
if (l) Min(t[k].mn[i],t[l].mn[i]),Max(t[k].mx[i],t[l].mx[i]);
if (r) Min(t[k].mn[i],t[r].mn[i]),Max(t[k].mx[i],t[r].mx[i]);
}
t[k].sum=t[k].v+t[l].sum+t[r].sum;
}
void ins(int &k,bool D){
if (!k){
k=++cnt;
t[k][0]=t[k].mn[0]=t[k].mx[0]=now[0];
t[k][1]=t[k].mn[1]=t[k].mx[1]=now[1];
}
if (now==t[k]){
t[k].v+=now.v;t[k].sum+=now.v;
return;
}
if (now[D]<t[k][D]) ins(t[k].l,D^1);
else ins(t[k].r,D^1);
update(k);
}
int query(int k,int x1,int y1,int x2,int y2){
if (!k) return 0;
int tmp=0;
if (in(x1,y1,x2,y2,t[k].mn[0],t[k].mn[1],t[k].mx[0],t[k].mx[1])) return t[k].sum;
if (out(x1,y1,x2,y2,t[k].mn[0],t[k].mn[1],t[k].mx[0],t[k].mx[1])) return 0;
if (in(x1,y1,x2,y2,t[k][0],t[k][1],t[k][0],t[k][1])) tmp+=t[k].v;
tmp+=query(t[k].l,x1,y1,x2,y2)+query(t[k].r,x1,y1,x2,y2);
return tmp;
}
int build(int l,int r,bool f){
if (l>r) return 0;
int mid=l+r>>1;
D=f;nth_element(p+l,p+mid,p+r+1);
t[mid]=p[mid];
t[mid].l=build(l,mid-1,f^1);
t[mid].r=build(mid+1,r,f^1);
update(mid);
return mid;
}
}T;
int main(){
n=read();m=10000;
while (1){
opt=read();if (opt==3) break;
x=read()^ans;y=read()^ans;
if (opt==1){
A=read()^ans;T.now[0]=x;T.now[1]=y;
T.now.v=T.now.sum=A;
T.ins(T.rt,0);
if (T.cnt==m){
for (i=1;i<=T.cnt;i++) p[i]=T.t[i];
T.rt=T.build(1,T.cnt,0);m+=10000;
}
}else{
xx=read()^ans;yy=read()^ans;
wln(ans=T.query(T.rt,x,y,xx,yy));
}
}
}