題意:有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;
}