[BZOJ4066]简单题（kd-tree）

题目描述

题解

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 200005

int s,opt,n,x,y,v,p,q,cmpd,root,ans;
struct data
{
int l,r,val,sum;
int d[2],mx[2],mn[2];
}tr
;

void update(int x)
{
int l=tr[x].l,r=tr[x].r;
tr[x].sum=tr[x].val;
if (l)
{
tr[x].mn[0]=min(tr[x].mn[0],tr[l].mn[0]);
tr[x].mx[0]=max(tr[x].mx[0],tr[l].mx[0]);
tr[x].mn[1]=min(tr[x].mn[1],tr[l].mn[1]);
tr[x].mx[1]=max(tr[x].mx[1],tr[l].mx[1]);
tr[x].sum+=tr[l].sum;
}
if (r)
{
tr[x].mn[0]=min(tr[x].mn[0],tr[r].mn[0]);
tr[x].mx[0]=max(tr[x].mx[0],tr[r].mx[0]);
tr[x].mn[1]=min(tr[x].mn[1],tr[r].mn[1]);
tr[x].mx[1]=max(tr[x].mx[1],tr[r].mx[1]);
tr[x].sum+=tr[r].sum;
}
}
int cmp(data a,data b)
{
return a.d[cmpd]<b.d[cmpd]||a.d[cmpd]==b.d[cmpd]&&a.d[cmpd^1]<b.d[cmpd^1];
}
int rebuild(int l,int r,int d)
{
int mid=(l+r)>>1;
cmpd=d;
nth_element(tr+l,tr+mid,tr+r+1,cmp);
tr[mid].mx[0]=tr[mid].mn[0]=tr[mid].d[0];
tr[mid].mx[1]=tr[mid].mn[1]=tr[mid].d[1];
tr[mid].sum=tr[mid].val;
tr[mid].l=tr[mid].r=0;
if (l<mid) tr[mid].l=rebuild(l,mid-1,d^1);
if (mid<r) tr[mid].r=rebuild(mid+1,r,d^1);
update(mid);
return mid;
}
void insert(int x)
{
if (!root)
{
root=x;
return;
}
int now=root,d=0;
while (1)
{
tr[now].mn[0]=min(tr[now].mn[0],tr[x].mn[0]);
tr[now].mx[0]=max(tr[now].mx[0],tr[x].mx[0]);
tr[now].mn[1]=min(tr[now].mn[1],tr[x].mn[1]);
tr[now].mx[1]=max(tr[now].mx[1],tr[x].mx[1]);
tr[now].sum+=tr[x].sum;
if (tr[x].d[d]<=tr[now].d[d])
{
if (tr[now].l) now=tr[now].l;
else {tr[now].l=x;break;}
}
else
{
if (tr[now].r) now=tr[now].r;
else {tr[now].r=x;break;}
}
d^=1;
}
}
bool ok(int id)
{
return (tr[id].d[0]>=x&&tr[id].d[0]<=p&&tr[id].d[1]>=y&&tr[id].d[1]<=q);
}
int check(int id)
{
if (tr[id].mn[0]>=x&&tr[id].mx[0]<=p&&tr[id].mn[1]>=y&&tr[id].mx[1]<=q) return 1;
else if (tr[id].mx[0]<x||tr[id].mn[0]>p||tr[id].mx[1]<y||tr[id].mn[1]>q) return -1;
else return 0;
}
void query(int now)
{
if (ok(now)) ans+=tr[now].val;
if (tr[now].l)
{
int t=check(tr[now].l);
if (t==1) ans+=tr[tr[now].l].sum;
else if (t==0) query(tr[now].l);
}
if (tr[now].r)
{
int t=check(tr[now].r);
if (t==1) ans+=tr[tr[now].r].sum;
else if (t==0) query(tr[now].r);
}
}
int main()
{
scanf("%d",&s);
while (~scanf("%d",&opt))
{
if (opt==3) break;
if (opt==1)
{
scanf("%d%d%d",&x,&y,&v);++n;
x^=ans,y^=ans,v^=ans;
tr
.mx[0]=tr
.mn[0]=tr
.d[0]=x;
tr
.mx[1]=tr
.mn[1]=tr
.d[1]=y;
tr
.val=tr
.sum=v;
insert(n);
if (n%10000==0) root=rebuild(1,n,0);
}
else
{
scanf("%d%d%d%d",&x,&y,&p,&q);
x^=ans,y^=ans,p^=ans,q^=ans;
ans=0;
query(root);
printf("%d\n",ans);
}
}
return 0;
}