求出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; }