POJ 2186 Popular Cows(强连通分量)

题目分析

就是找所有牛都认为是红人的牛的个数，一看就知道是强连通分量按拓扑排序之后标号最大的那个，然后根据反向遍扫一遍看看是否能够把所有点都跑一边，如果可以，说明编号最大的那个强连通分量中所有节点都可以被所有人都看做红人。

#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 1e4+100;
int N, M;
vector <int> G[maxn];
vector <int> rG[maxn];
vector <int> vs;
int cmp[maxn];
bool used[maxn];

void addedge(int from, int to){
    G[from].push_back(to);
    rG[to].push_back(from);
}

void dfs(int u){
    used[u] = true;
    for(int i = 0; i < G[u].size(); i++)
        if(!used[G[u][i]]) dfs(G[u][i]);
    vs.push_back(u);
}

void rdfs(int u, int k){
    used[u] = true;
    cmp[u] = k;
    for(int i = 0; i < rG[u].size(); i++)
        if(!used[rG[u][i]]) rdfs(rG[u][i], k);
}

int scc(){
    memset(used, 0, sizeof(used));
    vs.clear();
    for(int i = 1; i <= N; i++)
        if(!used[i]) dfs(i);
    int k = 0;
    memset(used, 0, sizeof(used));
    for(int i = vs.size()-1; i >= 0; i--)
        if(!used[vs[i]]) rdfs(vs[i], k++);
    return k;
}

void init(){
    for(int i = 0; i < maxn; i++)
        G[i].clear(), rG[i].clear();
    vs.clear();
}

void solve(){
    int k = scc();
    int ans = 0, u = 0;
    for(int i = 1; i <= N; i++)
        if(cmp[i] == k-1) u = i,ans++;
    memset(used, 0, sizeof(used));
    rdfs(u, 0);
    for(int i = 1; i <= N; i++)
        if(!used[i]){ ans = 0; break; }
    printf("%d\n", ans);
}

int main(){
    while(scanf("%d%d", &N, &M) != EOF){
        init();
        int from, to;
        for(int i = 0; i < M; i++){
            scanf("%d%d", &from, &to);
            addedge(from, to);
        }
        solve();
    }
    return 0;
}