BZOJ1051: [HAOI2006]受欢迎的牛

题解:

这题是比较典型的tarjan缩环问题，只要刷一次tarjan就可以刷出答案。
代码如下:

#include<cstdio>
#include<string>
using namespace std;
const int maxn=100005,maxm=500005;
int n,e,lnk[maxn],son[maxm],nxt[maxm],tot,dfn[maxn],low[maxn],stark[maxn],top,ti,fa[maxn],sum[maxn],ans,ans1;
bool vis[maxn],vis1[maxn];
void add(int x,int y) {
    son[++tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;
}
void tarjan(int x){
    dfn[x]=low[x]=++ti; stark[++top]=x; vis1[x]=1;
    for (int j=lnk[x];j;j=nxt[j])
    if (!dfn[son[j]]) tarjan(son[j]),low[x]=min(low[x],low[son[j]]); else if (vis1[son[j]]) low[x]=min(low[x],dfn[son[j]]);
    if (dfn[x]==low[x]) do{vis1[stark[top]]=0; sum[fa[stark[top]]=x]++;} while (stark[top--]!=x);
}
int main(){
    scanf("%d %d",&n,&e);
    for (int i=1;i<=n;i++) fa[i]=i;
    for (int i=1;i<=e;i++) {
        int x,y; scanf("%d %d",&x,&y); add(x,y);
    }
    for (int i=1;i<=n;i++) if (!dfn[i]) tarjan(i);
    for (int i=1;i<=n;i++)
    for (int j=lnk[i];j;j=nxt[j])
    if (fa[son[j]]^fa[i]) vis[fa[i]]=1;
    for (int i=1;i<=n;i++)
    if (sum[fa[i]]) {
        if (!vis[fa[i]]) ans1++,ans=sum[fa[i]];
        sum[fa[i]]=0;
    }
    if (ans1>1) printf("0\n"); else printf("%d\n",ans);
    return 0;
}