bzoj2815 [ZJOI2012]灾难 支配树

没事学习一下新算法= =，反正也不是很难，WC讲过的拖到现在真是不好意思啊。
个人感觉基本上恨少考这个东西= =
题意：给你一个DAG，对于每一个点去掉以后有多少个点不能从入度为0的点到达。
那么这明显就是支配树，只要我删的这个点是必经点，那么我在支配树上的所有儿子都不能到达，那么每个点的答案就是他在支配树的上的size-1。
建支配树用lca，注意只有dag才能用这个方法。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
queue<int>q;
int head[N],head1[N],n,next[N],go[N],val[N];
int fa[N][20],dep[N],d[N],size[N],a[N],tot;
inline void add(int x,int y)
{
    go[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
    go[++tot]=x;
    next[tot]=head1[y];
    head1[y]=tot;
}
inline void topsort()
{
    q.push(n+1);
    int tot=0;
    while (!q.empty())
    {
        int x=q.front();
        q.pop(),a[++tot]=x;
        for(int i=head[x];i;i=next[i])
        {
            int v=go[i];
            d[v]--;
            if (!d[v])q.push(v);
        }
    }
}
int get_lca(int x,int y)
{
    if (dep[x]<dep[y]) swap(x,y);
    for (int i=16;i>=0;i--)
        if (dep[fa[x][i]]>=dep[y]) x=fa[x][i];
    if (x==y) return x;
    for (int i=16;i>=0;i--)
        if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}

int main()
{
    scanf("%d",&n);
    fo(i,1,n)
    {
        int x;
        scanf("%d",&x);
        while (x)add(x,i),scanf("%d",&x),d[i]++;
    }
    fo(i,1,n)
    if (!d[i])add(n+1,i),d[i]++;
    topsort();
    fo(i,1,n+1)
    {
        int x=a[i],y=0;
        for(int j=head1[x];j;j=next[j])
        {
            if (!y)y=go[j];
            else y=get_lca(go[j],y);
        }
        fa[x][0]=y;
        dep[x]=dep[y]+1;
        fo(j,1,16)fa[x][j]=fa[fa[x][j-1]][j-1];
    }
    fd(i,n+1,1)size[a[i]]++,size[fa[a[i]][0]]+=size[a[i]];
    fo(i,1,n)printf("%d\n",size[i]-1);
}