[kosaraju算法] POJ 2186 Popular Cows

题面

link
  给定一个有向图，问有几个结点，图中的任意结点都可以到达？($1\le n\le 1e4$)

分析

  若是知道了图中的强连通分量，将一个强连通分量缩成一个点的话，那么原图可以变成是一个DAG(有向无环图)。若是只有一个结点出度为0，那么其它结点都可以到达这个点；若是有大于一个，那么就不存在任何结点都可以到达的点了。
  所以我们可以用 kosaraju 算法求出强连通分量，进行缩点，然后求出出度为0的缩点判断答案。
  kosaraju算法的步骤是：①正向建图和反向建图；②遍历正(反)向图，记录后序(最先遍历到的结点在后面)；③根据后序遍历反(正)向图，每次dfs都可以遍历到一个强连通分量。
  kosaraju算法的核心是先通过一次 dfs 得到合适的遍历顺序，然后第二次 dfs 得到强连通分量，觉得知乎上一个很好的讲解：link (用食物链关系来解释这个算法过程)

代码

#include <vector>
#include <stdio.h>
#define pb push_back

using namespace std;

typedef long long ll;
typedef pair<int, int> P;
const int maxn = 1e4 + 10;
const int INF = 0x3f3f3f3f;
const ll mod = 1e9 + 7;

int n, m, vis[maxn], st[maxn], mark[maxn], num[maxn], d[maxn], tol, tag;
vector<int> G1[maxn], G2[maxn];

void dfs1(int x)
{
    vis[x] = 1;
    for(unsigned int i = 0; i < G1[x].size(); i++)
    {
        int y = G1[x][i];
        if(vis[y])  continue;
        dfs1(y);
    }
     st[++tol] = x;       //最先遍历到的结点在栈顶
}

void dfs2(int x)
{
    vis[x] = 1;
    for(unsigned int i = 0; i < G2[x].size(); i++)
    {
        int y = G2[x][i];
        if(vis[y])  continue;
        dfs2(y);
    }
    mark[x] = tag;            //同一个dfs遍历到的结点属于同一个强连通分量
    num[tag]++;
}

void kosaraju()
{
    for(int i = 1; i <= n; i++)
        if(!vis[i])
            dfs1(i);

    fill(vis + 1, vis + 1 + n, 0);
    for(int i = tol; i >= 1; i--)
        if(!vis[st[i]])
        {
            tag++;
            dfs2(st[i]);
        }

    for(int i = 1; i <= n; i++)
    {
        for(unsigned int j = 0; j < G1[i].size(); j++)
        {
            int y = G1[i][j];
            if(mark[i] == mark[y])  continue;
            d[mark[i]]++;                   //标记连通分量对应点的出度
        }
    }

    int cnt = 0, ans;
    for(int i = 1; i <= tag; i++)
    {
        if(!d[i])
        {
            if(cnt++ == 1)
            {
                printf("0\n");
                return;
            }
            ans = num[i];
        }
    }
    printf("%d\n", ans);
}

int main()
{
    scanf("%d %d", &n, &m);
    for(int i = 1; i <= m; i++)
    {
        int x, y;
        scanf("%d %d", &x, &y);
        G1[x].pb(y);  G2[y].pb(x);    //正向建图和反向建图
    }
    kosaraju();
}