[BZOJ4066]简单题（kd-tree）

题目描述

传送门

题解

有强制在线，不能用cdq分治做了= =
不过可以用kd-tree
由于是要求和，维护一个sum
如果要是左右子树都在范围内，直接加和
如果都不在范围内，跳过
否则递归子树

代码

#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[N];

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[n].mx[0]=tr[n].mn[0]=tr[n].d[0]=x;
            tr[n].mx[1]=tr[n].mn[1]=tr[n].d[1]=y;
            tr[n].val=tr[n].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;
}