BZOJ4154 : [Ipsc2015]Generating Synergy

求出dfs序和每个点的深度

将第i个点看成二维点(st[i],d[i])

则修改操作等价于将横坐标在[st[x],en[x]]内，纵坐标在[d[x],d[x]+y]范围内的点的颜色都修改为c

用支持标记下放的k-d树维护即可，时间复杂度$O(n\log n+q\sqrt{n})$。

#include<cstdio>
#include<algorithm>
#define N 100010
int T,n,c,m,x,y,i,g[N],nxt[N],st[N],en[N],dfn,d[N],id[N],tmp[N],ans,root,cmp_d,x1,y1,x2,y2;
inline void read(int&a){char c;while(!(((c=getchar())>=‘0‘)&&(c<=‘9‘)));a=c-‘0‘;while(((c=getchar())>=‘0‘)&&(c<=‘9‘))(a*=10)+=c-‘0‘;}
struct node{int d[2],l,r,Max[2],Min[2],tag,col,f;}t[N];
inline bool cmp(node a,node b){return a.d[cmp_d]<b.d[cmp_d];}
void dfs(int x){
  t[x].tag=x,t[x].d[0]=st[x]=++dfn,t[x].d[1]=d[x];
  for(int i=g[x];i;i=nxt[i])d[i]=d[x]+1,dfs(i);
  en[x]=dfn;
}
inline void umax(int&a,int b){if(a<b)a=b;}
inline void umin(int&a,int b){if(a>b)a=b;}
inline void up(int x){
  id[t[x].tag]=x,t[x].tag=0,t[x].col=1;
  if(t[x].l){
    umax(t[x].Max[0],t[t[x].l].Max[0]);
    umin(t[x].Min[0],t[t[x].l].Min[0]);
    umax(t[x].Max[1],t[t[x].l].Max[1]);
    umin(t[x].Min[1],t[t[x].l].Min[1]);
  }
  if(t[x].r){
    umax(t[x].Max[0],t[t[x].r].Max[0]);
    umin(t[x].Min[0],t[t[x].r].Min[0]);
    umax(t[x].Max[1],t[t[x].r].Max[1]);
    umin(t[x].Min[1],t[t[x].r].Min[1]);
  }
}
int build(int l,int r,int D,int f){
  int mid=(l+r)>>1;
  cmp_d=D,std::nth_element(t+l+1,t+mid+1,t+r+1,cmp);
  t[mid].f=f;
  t[mid].Max[0]=t[mid].Min[0]=t[mid].d[0];
  t[mid].Max[1]=t[mid].Min[1]=t[mid].d[1];
  if(l!=mid)t[mid].l=build(l,mid-1,!D,mid);else t[mid].l=0;
  if(r!=mid)t[mid].r=build(mid+1,r,!D,mid);else t[mid].r=0;
  return up(mid),mid;
}
inline void tag1(int x,int y){t[x].col=t[x].tag=y;}
inline void pb(int x){
  if(t[x].tag){
    if(t[x].l)tag1(t[x].l,t[x].tag);
    if(t[x].r)tag1(t[x].r,t[x].tag);
    t[x].tag=0;
  }
}
inline void change(int x){
  if(t[x].Max[0]<x1||t[x].Min[0]>x2||t[x].Max[1]<y1||t[x].Min[1]>y2)return;
  if(t[x].Min[0]>=x1&&t[x].Max[0]<=x2&&t[x].Min[1]>=y1&&t[x].Max[1]<=y2){tag1(x,c);return;}
  pb(x);
  if(t[x].d[0]>=x1&&t[x].d[0]<=x2&&t[x].d[1]>=y1&&t[x].d[1]<=y2)t[x].col=c;
  if(t[x].l)change(t[x].l);
  if(t[x].r)change(t[x].r);
}
inline int ask(int x){
  int s=0,i=x;
  while(t[i].f)tmp[++s]=i=t[i].f;
  while(s)pb(tmp[s--]);
  return t[x].col;
}
int main(){
  for(read(T);T--;printf("%d\n",ans)){
    read(n),read(c),read(m);
    for(i=1;i<=n;i++)g[i]=0;
    for(i=2;i<=n;i++)read(x),nxt[i]=g[x],g[x]=i;
    dfs(1),root=build(1,n,0,0),ans=0;
    for(i=1;i<=m;i++){
      read(x),read(y),read(c);
      if(c)x1=st[x],x2=en[x],y1=d[x],y2=d[x]+y,change(root);
      else ans=(1LL*ask(id[x])*i+ans)%1000000007;
    }
  }
  return 0;
}