BZOJ 4154 [Ipsc2015]Generating Synergy(KD-Tree)

题目链接：BZOJ 4154 [Ipsc2015]Generating Synergy

题意：

给定一棵以1为根的有根树,初始所有节点颜色为1,每次将距离节点a不超过l的a的子节点染成c,或询问点a的颜色。

题解：

将dfs序看为x，dep看为y，那么就是一个在二维平面上的操作了。

由于这个平面范围比较大，二维线段树不好开，然后kd-tree搞搞。

#include<cstdio>
#include<algorithm>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

namespace KD_Tree{
    const int N=1e5+7,DI=2;
    struct Node{
        int p[DI],f,l,r,mx[DI],mi[DI],tim,col,t,c;
        int operator[](const int&idx)const{return p[idx];}
        void in(int idx,int *v){f=idx,tim=t=0,c=col=1;F(i,0,DI-1)p[i]=v[i];}
        void up(Node&a){
            F(i,0,DI-1){
                mi[i]=min(mi[i],a.mi[i]);
                mx[i]=max(mx[i],a.mx[i]);
            }
        }
    }T[N];
    int idx[N],cmpd,root;
    
    bool cmp(const Node&a,const Node&b){return a[cmpd]<b[cmpd];}
    void up(int x){
        if(T[x].l)T[x].up(T[T[x].l]);
        if(T[x].r)T[x].up(T[T[x].r]);
    }
    int build(int l,int r,int d=0,int f=0)
    {
        int mid=l+r>>1;
        cmpd=d%DI,nth_element(T+l,T+mid,T+r+1,cmp);
        idx[T[mid].f]=mid,T[mid].f=f;
        F(i,0,DI-1)T[mid].mi[i]=T[mid].mx[i]=T[mid][i];
        T[mid].l=l!=mid?build(l,mid-1,d+1,mid):0;
        T[mid].r=r!=mid?build(mid+1,r,d+1,mid):0;
        return up(mid),mid;
    }
    void update(int x1,int x2,int y1,int y2,int t,int c,int rt=root)
    {
        if(T[rt].mi[0]>=x1&&T[rt].mx[0]<=x2&&T[rt].mi[1]>=y1&&T[rt].mx[1]<=y2)
        {
            T[rt].tim=t,T[rt].col=c;
            T[rt].t=t,T[rt].c=c;
            return;
        }
        if(T[rt].mx[0]<x1||T[rt].mi[0]>x2||T[rt].mx[1]<y1||T[rt].mi[1]>y2)
            return;
        if(T[rt][0]>=x1&&T[rt][0]<=x2&&T[rt][1]>=y1&&T[rt][1]<=y2)
            T[rt].tim=t,T[rt].col=c;
        if(T[rt].l)update(x1,x2,y1,y2,t,c,T[rt].l);
        if(T[rt].r)update(x1,x2,y1,y2,t,c,T[rt].r);
    }
    int ask(int rt)
    {
        int t=T[rt].tim,c=T[rt].col;
        while(T[rt].f)
        {
            rt=T[rt].f;
            if(T[rt].t>t)t=T[rt].t,c=T[rt].c;
        }
        return c;
    }
};
const int N=1e5+7,P=1e9+7;
int t,n,q,g[N],v[N],nxt[N],ed,c;
int x,in[N],out[N],cnt,dep[N],ans;

void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}

void dfs(int x,int d=0)
{
    in[x]=++cnt,dep[x]=d;
    for(int i=g[x];i;i=nxt[i])dfs(v[i],d+1);
    out[x]=++cnt;
}

using namespace KD_Tree;
int main(){
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&c,&q);
        F(i,1,n)g[i]=0;ed=0;
        F(i,2,n)scanf("%d",&x),adg(x,i);
        ans=0,cnt=0,dfs(1);
        F(i,1,n)
        {
            int x[2];
            x[0]=in[i],x[1]=dep[i];
            T[i].in(i,x);
        }
        root=build(1,n);
        F(i,1,q)
        {
            int a,l,c;
            scanf("%d%d%d",&a,&l,&c);
            if(c)update(in[a],out[a],dep[a],dep[a]+l,i,c);
            else
            {
                int tmp=ask(idx[a]);
                ans=(ans+1ll*tmp*i)%P;
            }
        }
        printf("%d
",ans);
    }
    return 0;
}