bzoj2815 [ZJOI2012]灾难 支配树

Description

---

阿米巴是小强的好朋友。

阿米巴和小强在草原上捉蚂蚱。小强突然想，如果蚂蚱被他们捉灭绝了，那么吃蚂蚱的小鸟就会饿死，而捕食小鸟的猛禽也会跟着灭绝，从而引发一系列的生态灾难。

学过生物的阿米巴告诉小强，草原是一个极其稳定的生态系统。如果蚂蚱灭绝了，小鸟照样可以吃别的虫子，所以一个物种的灭绝并不一定会引发重大的灾难。

我们现在从专业一点的角度来看这个问题。我们用一种叫做食物网的有向图来描述生物之间的关系：

一个食物网有N个点，代表N种生物，如果生物x可以吃生物y，那么从y向x连一个有向边。

这个图没有环。

图中有一些点没有连出边，这些点代表的生物都是生产者，可以通过光合作用来生存； 而有连出边的点代表的都是消费者，它们必须通过吃其他生物来生存。

如果某个消费者的所有食物都灭绝了，它会跟着灭绝。

我们定义一个生物在食物网中的“灾难值”为，如果它突然灭绝，那么会跟着一起灭绝的生物的种数。

举个例子：在一个草场上，生物之间的关系是：

如

如果小强和阿米巴把草原上所有的羊都给吓死了，那么狼会因为没有食物而灭绝，而小强和阿米巴可以通过吃牛、牛可以通过吃草来生存下去。所以，羊的灾难值是1。但是，如果草突然灭绝，那么整个草原上的5种生物都无法幸免，所以，草的灾难值是4。

给定一个食物网，你要求出每个生物的灾难值。

对50%的数据，N ≤ 10000。
对100%的数据，1 ≤ N ≤ 65534。
输入文件的大小不超过1M。保证输入的食物网没有环。

Solution

---

考虑把非生物因素作为新的根连接生产者，我们建出新图的支配树
答案显然为支配树上的size-1

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)

const int N=500005;

namespace Tree {
    struct edge {int y,next;} e[N];

    int size[N];
    int ls[N],edCnt;

    void add_edge(int x,int y) {
        e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
    }

    void dfs(int now) {
        size[now]=1;
        for (int i=ls[now];i;i=e[i].next) {
            dfs(e[i].y); size[now]+=size[e[i].y];
        }
    }
} ;

struct edge {int y,next;} e[N];

std:: vector <int> pre[N],dom[N];

int dfn[N],id[N],fa[N],acs[N],mn[N];
int idom[N],sdom[N];
int ls[N],edCnt;

int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}

void add_edge(int x,int y) {
    e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
}

void dfs(int now) {
    dfn[now]=++dfn[0];
    id[dfn[0]]=now;
    for (int i=ls[now];i;i=e[i].next) {
        pre[e[i].y].push_back(now);
        if (!dfn[e[i].y]) {
            fa[e[i].y]=now;
            dfs(e[i].y);
        }
    }
}

int find(int x) {
    if (acs[x]==x) return x;
    int tmp=find(acs[x]);
    if (dfn[sdom[mn[acs[x]]]]<dfn[sdom[mn[x]]]) mn[x]=mn[acs[x]];
    return acs[x]=tmp;
}

int smin(int x,int y) {
    return (dfn[x]<dfn[y])?x:y;
}

void build(int n) {
    drp(i,dfn[0],1) {
        int x=id[i];
        if (!pre[x].empty()) {
            for (int j=0;j<pre[x].size();j++) {
                int y=pre[x][j];
                if (dfn[y]<dfn[x]) sdom[x]=smin(sdom[x],y);
                else {
                    find(y);
                    sdom[x]=smin(sdom[x],sdom[mn[y]]);
                }
            }
            pre[x].clear();
        }
        acs[x]=fa[x];
        dom[sdom[x]].push_back(x);
        if (!dom[fa[x]].empty()) {
            for (int j=0;j<dom[fa[x]].size();j++) {
                int y=dom[fa[x]][j];
                find(y); int d=mn[y];
                if (dfn[sdom[d]]>=dfn[sdom[y]]) idom[y]=sdom[y];
                else idom[y]=d;
            }
            dom[fa[x]].clear();
        }
    }
    rep(i,1,dfn[0]) {
        int x=id[i];
        if (idom[x]!=sdom[x]) idom[x]=idom[idom[x]];
        Tree:: add_edge(idom[x],x);
    }
    Tree:: dfs(n);
    rep(i,1,n-1) printf("%d\n", Tree:: size[i]-1);
}

int main(void) {
    int n=read();
    rep(i,1,n) {
        int x=read();
        if (!x) add_edge(n+1,i);
        else for (;x;x=read()) add_edge(x,i);
    }
    rep(i,1,n+1) sdom[i]=acs[i]=mn[i]=i;
    dfs(n+1);
    build(n+1);
    return 0;
}