题目
题面
建立出原图的支配树的图。以一个点为根的子树的大小-1(减去该点)就是该点的伤害值。
模板题 不说啦。
#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
#define en '\n
#define m(a,b) memset(a,b,sizeof a)
using namespace std;
typedef long long ll;
const int N=1e5+10,M=1e6+10;
struct Edge{int to,nex;};
struct Graph{
Edge e[M];int head[N],tot;
void init(){tot=0;m(head,0);}
void add(int from,int to){
e[++tot]=(Edge){to,head[from]};head[from]=tot;
}
}A,B,C;
int dep[N],fa[N][25],out[N],a[N],root,K=20;
queue<int>q;
int LCA(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int j=K;j>=0;j--)
if(dep[x]-(1<<j)>=dep[y]) x=fa[x][j];
if(x==y) return x;
for(int j=K;j>=0;j--)
if(fa[x][j]!=fa[y][j]) x=fa[x][j],y=fa[y][j];
return fa[x][0];
}
int n;
void build(){
int num=0;root=n+1;
for(int i=1;i<=n;++i)
if(!out[i]) q.push(i),A.add(root,i),B.add(i,root);
while(!q.empty()){
int x=q.front();q.pop();
a[++num]=x;
for(int i=A.head[x];i;i=A.e[i].nex){
int y=A.e[i].to;
if(!(--out[y])) q.push(y);
}
}
dep[root]=1;
for(int i=1;i<=n;++i){
int x=a[i],bf=-1;
for(int j=B.head[x];j;j=B.e[j].nex){
int y=B.e[j].to;
bf=(bf==-1)?y:LCA(bf,y);
}
dep[x]=dep[bf]+1,fa[x][0]=bf,C.add(fa[x][0],x);
for(int i=1;i<=K;++i) fa[x][i]=fa[fa[x][i-1]][i-1];
}
}
int ans[N];
void dfs(int x){
ans[x]=1;
for(int i=C.head[x];i;i=C.e[i].nex){
int y=C.e[i].to;
dfs(y),ans[x]+=ans[y];
}
}
int main(){
//m(fa,0),m(out,0);
int x;scanf("%d",&n);
for(int i=1;i<=n;++i)
while(~scanf("%d",&x),x) A.add(x,i),B.add(i,x),++out[i];
build();
dfs(root);
for(int i=1;i<=n;++i) printf("%d\n",ans[i]-1);
}