POJ - 3180 The Cow Prom （ korasaju 算法模板）

The Cow Prom POJ - 3180

题意：

奶牛圆舞：N头牛，M条有向绳子，能组成几个歌舞团（团内奶牛数 n >= 2)？要求顺时针逆时针都能带动舞团内所有牛。

分析：

所谓能带动，就是舞团构成一个强连通分量，就是赤裸裸的SCC。

代码实现：很好的一道题，有利于理解 korasaju 算法

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
#define ms(a,b) memset(a,b,sizeof a)
const int maxn = 1e5 + 10;

int V;                      // 顶点数
vector<int> G[maxn];       // 图的邻接表表示
vector<int> rG[maxn];      // 反向图
vector<int> vs;             // 后序遍历顺序的顶点列表
bool book[maxn];           // 访问标记
int cmp[maxn];             // 所属强连通分量的拓补序

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

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

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

int scc() {
	ms(book, false); vs.clear();
	for (int v = 0; v < V; ++v)
		if (!book[v])dfs(v);
	ms(book, false); int k = 0;
	for (int i = vs.size() - 1; i >= 0; --i)
		if (!book[vs[i]])rdfs(vs[i], k++);
	//cout << k << endl; //3个连通分量
	return k;
}

int main() {
	//freopen("in.txt","r",stdin);
	ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	int m; cin >> V >> m;
	for (int i = 0; i < m; ++i) {
		int from, to; cin >> from >> to;
		add_edge(--from, --to);
	}
	int n = scc();
	vector<int>count(n, 0);
	for (int v = 0; v < V; ++v) { 
		//cout << cmp[v] << " "; // 2 2 1 2 0
		++count[cmp[v]]; 
	}
	//cout << endl;
	int ans = 0;
	for (int i = 0; i < n; ++i)
		if (count[i] >= 2)++ans;
	cout << ans << endl;
	return 0;
}