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
,nxt
,st
,en
,dfn,d
,id
,tmp
,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
;
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;
}