洛谷 P2597 [ZJOI2012]灾难（拓扑排序+lca）

一道很有意思的题目。
首先看到这种题目很容易想到拓扑排序。然后我们考虑建树，让食物成为消费者的父亲，这样食物灭绝其子树就会灭绝。但一个消费者有多个食物，让谁成为它的父亲？我们假设按拓扑序建树，那么加入一个结点u的时候，其食物一定全部在树上了。u灭绝的条件是其所有食物灭绝，也就等价于所有食物的lca灭绝，因此让lca作为u的父亲即可。
注意：可能存在多个生产者，这样建出的树是有多个根的，因此我们加一个辅助点，让它成为所有生产者的食物，最后树的根即为这个辅助点。

#include <bits/stdc++.h>
using namespace std;
const int maxn=70000,maxm=300010;
int n;
vector <int> food[maxn];

inline int read(){
    int ret=0,sign=1;
    char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') sign=-1; ch=getchar();}
    while(ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=getchar();
    return ret*sign;
}

int head[maxn],Ecnt,deg[maxn];
struct edge{
    int to,nxt;
}G[maxm];
inline void addE(int u,int v){
    G[Ecnt]=(edge){v,head[u]};
    head[u]=Ecnt++;
}

int topo[maxn],Tcnt;
queue <int> Q;
void toposort(){
    topo[++Tcnt]=n+1,Q.push(n+1);
    while(!Q.empty()){
        int u=Q.front(); Q.pop();
        for(int i=head[u];i!=-1;i=G[i].nxt){
            int v=G[i].to;
            --deg[v];
            if(deg[v]<=0) topo[++Tcnt]=v,Q.push(v);
        }
    }
}

int fa[maxn][20],dep[maxn];
vector <int> son[maxn];
inline void init(int u){
    for(int i=1;i<=18;i++){
        if(fa[u][i-1]) fa[u][i]=fa[fa[u][i-1]][i-1];
        else break;
    }
}

inline int getlca(int u,int v){
    if(dep[u]<dep[v]) swap(u,v);
    for(int i=18;i>=0;i--)
        if(fa[u][i]&&dep[fa[u][i]]>=dep[v]) u=fa[u][i];
    if(u==v) return u;
    for(int i=18;i>=0;i--)
        if(fa[u][i]!=fa[v][i]) u=fa[u][i],v=fa[v][i];
    return fa[u][0];
}

void build(){
    dep[n+1]=1;
    for(int i=2;i<=n+1;i++){
        int u=topo[i],lca=food[u][0];
        for(int j=1;j<food[u].size();j++)
            lca=getlca(lca,food[u][j]);
        fa[u][0]=lca,son[lca].push_back(u),dep[u]=dep[lca]+1;
        init(u);
    }
}

int f[maxn];
void solve(int u){
    for(int i=0;i<son[u].size();i++){
        solve(son[u][i]);
        f[u]+=f[son[u][i]]+1;
    }
}

int main(){
    memset(head,-1,sizeof(head));
    n=read();
    for(int i=1;i<=n;i++)
        while(1){
            int x=read();
            if(x==0){
                if(!food[i].size()) food[i].push_back(n+1),addE(n+1,i),++deg[i];
                break;
            }
            food[i].push_back(x),addE(x,i),++deg[i];
        }
    toposort();
    build();
    solve(n+1);
    for(int i=1;i<=n;i++) printf("%d\n",f[i]);
    return 0;
}