POJ 2186(Popular Cows)

題意:有n個奶牛,它們之間互相膜拜,並且如果A膜拜B,B膜拜C則,A也膜拜C,求有多少頭奶牛被所有奶牛都膜拜;
思路:強連通分量+縮點,剛學SCC,這題是書上的例題,順手敲下熟悉下,用的是Kosaraju算法,這個算法比較明瞭;
對於這個題目,先求出強連通分量的個數,如果強連通分量個數只有一個,就輸出n,否則對其進行縮點,然後統計各點
縮完點之後的度數,若縮完點後度數爲0的點有1個,則答案就是這個強連通分量的奶牛個數,如果大於1個則說明至少有
兩羣奶牛不相互膜拜,答案爲0。
第一題強連通分量題目,貼個代碼

#include <set>
#include <map>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#define pub push_back
using namespace std;

const int N = 10010;
//vis:訪問數組 st:訪問時間數組 scc:各強連通分量編號數組 oe:出入度素組 sum:各強連通分量點的個數 
int vis[N], st[N], scc[N], oe[N], sum[N];
vector<int> v1[N], v2[N];//v1爲原圖,v2爲轉置圖,也叫逆圖 

void dfsOne(int to, int &sig){
	vis[to] = 1;
	for(int i = 0;i < v1[to].size();i++)
		if(!vis[v1[to][i]])
			dfsOne(v1[to][i], sig);
	st[sig++] = to;
}

void dfsTwo(int to, int sig){
	vis[to] = 1;
	scc[to] = sig;
	for(int i = 0;i < v2[to].size();i++)
		if(!vis[v2[to][i]])
			dfsTwo(v2[to][i], sig);
}

int Kosaraju(int n){
	int sig = 1, i;
	memset(scc, 0, sizeof(scc));
	memset(st, 0, sizeof(st));
	memset(vis, 0, sizeof(vis));
	for(i = 1;i <= n;i++)
		if(!vis[i])
			dfsOne(i, sig);
	memset(vis, 0, sizeof(vis));
	sig = 0;
	for(i = n;i >= 1;i--)
		if(!vis[st[i]])
			dfsTwo(st[i], sig++);
	return sig;
}

void solve(int n, int sig){
	memset(oe, 0, sizeof(oe));//出度 
	memset(sum, 0, sizeof(sum));//各連通分量點的數量
	//縮點 
	for(int i = 1;i <= n;i++){
		sum[scc[i] + 1]++;
		for(int j = 0;j < v1[i].size();j++)
			if(scc[v1[i][j]] != scc[i])
				oe[scc[i] + 1]++;
	} 
	int res = 0, cnt = 0;
	for(int i = 1;i <= sig;i++)
		if(oe[i] == 0){
			cnt++;
			res += sum[i];
		}
	printf("%d\n", cnt > 1?0:res);
}

int main(){
	int n, m;
	while(cin >> n >> m){
		int i, a, b;
		for(i = 0;i < m;i++){
			scanf("%d%d", &a, &b);
			v1[a].pub(b);
			v2[b].pub(a);
		}
		int tot = Kosaraju(n);
		if(tot == 1)
			printf("%d\n", n);
		else
			solve(n, tot);
		for(i = 1;i <= n;i++){
			v1[i].clear();
			v2[i].clear();
		}
	}
	return 0;
}

這個是用tarjan實現的,居然比kosaraju要慢。。估計寫挫了。。

#include <set>
#include <map>
#include <queue>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#define pub push_back
using namespace std;

const int N = 10010;
int oe[N], sum[N];

/********************Tarjan********************/
int dfn[N], low[N], idx, a, scc[N];
int vis[N], stack[N], tot;
vector<int> v[N];

int tarjanDFS(int to){
	vis[to] = 1;
	stack[++idx] = to;
	low[to] = dfn[to] = ++a;
	for(int i = 0;i < v[to].size();i++){
		if(!vis[v[to][i]]){
			tarjanDFS(v[to][i]);
			low[to] = min(low[to], low[v[to][i]]);
		}
		if(vis[v[to][i]] == 1)
			low[to] = min(dfn[v[to][i]], low[to]);
	}
 	if(dfn[to] == low[to]){
 		tot++;
 		int e;
	 	do{
		 	e = stack[idx--];
	 		scc[e] = tot;
			vis[e] = 2;
	 	}while(e != to);
	 }
}

int tarjan(int n){
	idx = 0, tot = 0, a = 0;
	memset(vis, 0, sizeof(vis));
	memset(dfn, 0, sizeof(dfn));
	for(int i = 1;i <= n;i++)
		if(!vis[i])
			tarjanDFS(i);
	return tot;
}
/*********************END**********************/

void solve(int n){
	tarjan(n);
	if(tot == 1){
		printf("%d\n", n);
		return ;
	}
	memset(oe, 0, sizeof(oe));//出度 
	memset(sum, 0, sizeof(sum));//各連通分量點的數量
	//縮點
	for(int i = 1;i <= n;i++){
		sum[scc[i]]++;
		for(int j = 0;j < v[i].size();j++)
			if(scc[v[i][j]] != scc[i])
				oe[scc[i]]++;
	} 
	int res = 0, cnt = 0;
	for(int i = 1;i <= tot;i++)
		if(oe[i] == 0){
			cnt++;
			res += sum[i];
		}
	printf("%d\n", cnt > 1?0:res);
}

int main(){
	int n, m;
	while(cin >> n >> m){
		int i, a, b;
		for(i = 0;i < m;i++){
			scanf("%d%d", &a, &b);
			v[a].pub(b);
		}
		solve(n);
		for(int i = 1;i <= n;i++)
			v[i].clear();
	}
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章