POJ2168,Tarjan算法求强连通度，并查集

题目链接：Popular Cows
本题求的是受所有的牛欢迎的牛的个数（每头牛都欢迎它自己）。
本题建立的图模型是有向图，在有向图中有这么一个定理：从任意一个点出发，必定到达终点，终点的出度为0。在一个强连通分量中，任意两个点可以相互抵达，也可以说任意两个点是等价的。那么这道题可以这样去解决：将每个连通分量当作一个整体，缩小成为一个点。如果整个图中仅有一点的出度为0，那么在这个点所在的连通分量之外的点都只能够到达这个唯一的“终点”，而连通分量中的点都可以相互到达，所以，能够被所有点访问的点的个数就是该连通分量中点的个数！但是，如果该图中有两个或者两个以上的点的出度为0，那么至少这两个点不能相互访问的，所以不存在题目中要求的点。
我在这里用的是并查集表示在同一个连通分量中。
详见代码：

/*************************************************************************
	> File Name: main.cpp
	> Author:Eagles 
	> Mail:None 
	> Created Time: 2018年09月18日 星期二 21时54分36秒
	> Description:POJ2168,tarjan算法，并查集 
 ************************************************************************/

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>

using namespace std;
#define N 100000

struct node
{
    int to;
    int nex;
}E[N*2];//表示边

struct d
{
    int par,out,num;//num表示点所在集合的点的数目
    int low,dfn;
    bool is_in;//是否在栈中
}dot[N];

int head[N];
int n,m,cnt;
int dep;
stack<int>s;

int find_par(int x)
{
    int fa=dot[x].par;

    if (fa != x)
    {
        fa=find_par(dot[x].par);
        dot[x].num+=dot[fa].num;
    }

    return dot[x].par=fa;
}

void unite(int x, int y)
{
    int fx=find_par(x);
    int fy=find_par(y);

    dot[fy].par=fx;

    dot[fx].num+=dot[fy].num;
}

void addEdge(int a, int b)
{
    E[cnt].to=b;
    E[cnt].nex=head[a];
    head[a]=cnt++;
}

void init()
{
    memset(head,-1,sizeof(head));
    dep=0;
    while(!s.empty())
        s.pop();
    for (int i=0; i<n; i++)
    {
        dot[i].par=i;
        dot[i].num=1;
        dot[i].is_in=false;
        dot[i].out=0;
        dot[i].low=-1;
        dot[i].dfn=-1;
    }

    for (int i=0; i<m; i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        addEdge(a-1,b-1);
    }
}

void tarjan(int u)
{
    dot[u].is_in=true;
    dot[u].low=dot[u].dfn=dep++;
    s.push(u);

    for (int i=head[u]; i!=-1; i=E[i].nex)
    {
        int v=E[i].to;

        if (dot[v].dfn==-1)
        {
            tarjan(v);
            dot[u].low=min(dot[u].low,dot[v].low);
        }
        else if (dot[v].is_in)
            dot[u].low=min(dot[u].low,dot[v].low);
    }

    if (dot[u].low==dot[u].dfn)
    {
        int v=s.top();
        s.pop();

        while (v != u)
        {
            dot[v].is_in=false;
            unite(u,v);
            v=s.top();
            s.pop();
        }
        
        dot[u].is_in=false;
    }
}

void solve()
{
    for (int i=0; i<n; i++)
    {
        if (dot[i].dfn==-1)
            tarjan(i);
    }

    for (int i=0; i<n; i++)
    {
        for (int j=head[i]; j!=-1; j=E[j].nex)
        {
            if (dot[i].par != dot[E[j].to].par)
                dot[dot[i].par].out++;
        }
    }

    int ans=0;

    for (int i=0; i<n; i++)
    {
        if (dot[i].par==i&&dot[i].out==0)
        {
            if (ans==0)
                ans=dot[i].num;
            else
                ans=-1;
        }
    }

    printf("%d\n",ans==-1?0:ans);
}

int main()
{
    while (~scanf("%d%d",&n,&m))
    {
        init();
        solve();
    }
    return 0;
}