BZOJ4448 SCOI2015 情報傳遞
阿新 • • 發佈:2018-11-11
題目大意
給定一棵樹,支援兩種操作:將一個點染黑,詢問路徑上在$k$步操作前就已經被染黑的點的數量。
題解
將染黑看做給一個點賦其操作編號點權,每次詢問路徑上點權小於一定值的數量。
這樣會有一個性質,由於$k>0$,所以我們完全可以在查詢之前將所有點權賦上。
所以只需要離線然後查詢就好了,可以在樹上利用差分主席樹維護。
#include<algorithm> #include<iostream> #include<cassert> #include<cstring> #include<cstdio> #include<cmath> #define LL long long #define M 300020 using namespace std; namespace IO{ const int BS=(1<<20); int Top=0; char Buffer[BS],OT[BS],*OS=OT,*HD,*TL,SS[20]; const char *fin=OT+BS-1; char Getchar(){if(HD==TL){TL=(HD=Buffer)+fread(Buffer,1,BS,stdin);} return *HD++;} void flush(){fwrite(OT,1,OS-OT,stdout);} void Putchar(char c){*OS++ =c;if(OS==fin)flush(),OS=OT;} void write(int x){ if(!x){Putchar('0');return;} while(x) SS[++Top]=x%10,x/=10; while(Top) Putchar(SS[Top]+'0'),--Top; } int read(){ int nm=0,fh=1; char cw=Getchar(); for(;!isdigit(cw);cw=Getchar()) if(cw=='-') fh=-fh; for(;isdigit(cw);cw=Getchar()) nm=nm*10+(cw-'0'); return nm*fh; } } using namespace IO; int n,m,rt[M],L[M*22],R[M*22],sum[M*22],fs[M],nt[M],to[M],tmp,cnt; int sz[M],mxs[M],tp[M],fa[M],dep[M],Root,u[M],v[M],c[M],T,col[M]; void link(int x,int y){nt[tmp]=fs[x],fs[x]=tmp,to[tmp++]=y;} void ins(int &x,int pre,int l,int r,int pos){ x=++cnt,L[x]=L[pre],R[x]=R[pre]; sum[x]=sum[pre]+1; if(l==r) return; int mid=((l+r)>>1); if(pos<=mid) ins(L[x],L[pre],l,mid,pos); else ins(R[x],R[pre],mid+1,r,pos); } int calc(int t1,int t2,int d1,int d2){return sum[t1]+sum[t2]-sum[d1]-sum[d2];} int qry(int t1,int t2,int d1,int d2,int l,int r,int pos){ int now=calc(t1,t2,d1,d2),mid=((l+r)>>1); if(l>=pos||!now) return 0; if(r<pos) return now; int ans1=qry(L[t1],L[t2],L[d1],L[d2],l,mid,pos); int ans2=qry(R[t1],R[t2],R[d1],R[d2],mid+1,r,pos); return ans1+ans2; } void dfs1(int x){ sz[x]=1,dep[x]=dep[fa[x]]+1; for(int i=fs[x];i!=-1;i=nt[i]){ dfs1(to[i]),sz[x]+=sz[to[i]]; if(sz[mxs[x]]<sz[to[i]]) mxs[x]=to[i]; } } void dfs2(int x,int dtp){ tp[x]=dtp,ins(rt[x],rt[fa[x]],1,T,col[x]);if(mxs[x]) dfs2(mxs[x],dtp); for(int i=fs[x];i!=-1;i=nt[i]) if(to[i]!=mxs[x]) dfs2(to[i],to[i]); } int lca(int x,int y){ while(tp[x]!=tp[y]){if(dep[tp[x]]<dep[tp[y]]) swap(x,y); x=fa[tp[x]];} return dep[x]<dep[y]?x:y; } int main(){ n=read(),memset(fs,-1,sizeof(fs)); for(int i=1;i<=n;i++) if(!(fa[i]=read())) Root=i;else link(fa[i],i); T=read(); for(int i=1;i<=n;i++) col[i]=T; for(int i=1;i<=T;i++){ int tpe=read(),x; if(tpe==2){x=read(),col[x]=min(col[x],i);continue;} m++,u[m]=read(),v[m]=read(),c[m]=read(),c[m]=i-c[m]; } dfs1(Root),dfs2(Root,Root); for(int i=1;i<=m;i++){ int x=u[i],y=v[i],k=lca(u[i],v[i]); write(dep[x]+dep[y]-(dep[k]<<1)+1),Putchar(' '); write(qry(rt[x],rt[y],rt[k],rt[fa[k]],1,T,c[i])),Putchar('\n'); } flush(); return 0; }