bzoj2815(DAG支配树模板)

题目
题面
建立出原图的支配树的图。以一个点为根的子树的大小-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);
}