BZOJ1051(HAOI2006)[受欢迎的牛]--并查集+Tarjan

【链接】
bzoj1051

【解题报告】

首先会考虑到会有森林，所以用并查集判断。

假设没有环的时候，如果要符合题意，那么图中只有一个出度为0的节点。

所以有环的时候，如果要符合题意，那么图中只有一个出度为0的环。

直接用Tarjan缩点再判断就好了。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=10005,maxm=50005;
int n,m,ans,tot,top,tme,fa[maxn],num[maxn],lnk[maxn],son[maxm],nxt[maxm],dfn[maxn],low[maxn],stk[maxn],f[maxn];
bool instk[maxn];
inline char nc()
{
    static char buf[100000],*l,*r;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
inline int Read()
{
    int res=0; char ch=nc();
    while (ch<'0'||ch>'9') ch=nc();
    while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=nc();
    return res;
}
void Add(int x,int y) {son[++tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;}
int Getfa(int x)
{
    if (x==fa[x]) return x;
    return fa[x]=Getfa(fa[x]);
}
void Tarjan(int x)
{
    dfn[x]=low[x]=++tme; instk[x]=1; stk[++top]=x; int y;
    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 (instk[son[j]]) low[x]=min(low[x],dfn[son[j]]);
    if (dfn[x]==low[x]) do {y=stk[top--]; instk[y]=0; fa[y]=x; num[x]++;} while(x!=y);
}
int main()
{
    freopen("1051.in","r",stdin);
    freopen("1051.out","w",stdout);
    n=Read(); m=Read(); top=tot=ans=0;
    memset(lnk,0,sizeof(lnk));
    for (int i=1; i<=n; i++) fa[i]=i;
    for (int i=1,x,y; i<=m; i++) x=Read(),y=Read(),Add(x,y),fa[Getfa(x)]=Getfa(y);
    for (int i=1; i<=n; i++) fa[i]=Getfa(fa[i]);
    for (int i=1; i<n; i++)
     if (fa[i]!=fa[i+1]) {printf("0\n"); return 0;}
    memset(dfn,0,sizeof(dfn));
    memset(num,0,sizeof(num));
    for (int i=1; i<=n; i++)
     if (!dfn[i]) Tarjan(i);
    memset(f,0,sizeof(f));
    for (int i=1; i<=n; i++)
     for (int j=lnk[i]; j; j=nxt[j])
      if (fa[i]!=fa[son[j]]) f[fa[i]]++;
    for (int i=1; i<=n; i++)
     if (i==fa[i]&&!f[i])
      if (ans) {printf("0\n"); return 0;} else ans=num[i];
    printf("%d\n",ans);
    return 0;
}