題意:有一羣牛,每對牛之間可以有崇拜關係,同時,崇拜關係滿足傳遞性。問被除自己以外的所有牛崇拜的牛有多少頭?
第一次做強連通分量的題。這道題想想其實很簡單。
思路:先按題意建圖,顯然,每個強聯通分量中的牛都是相互崇拜的,所以我們把所有強連通分量找出來,然後把它們都看成一個點。兩個強連通分量之間如果存在邊,則必存在邊(V1,V2)∈E,其中V1∈強連通分量1, V2∈強連通分量2。點有了,邊有了,新圖就出來了。然後找到這個新圖中度數爲0的點,如果有1個以上,則不可能存在一頭牛被所有牛崇拜。如果爲1,則輸出度數爲0的強連通分量中的點的數量,如果爲0,則輸出n,即所有牛都滿足條件。
代碼如下:
#include<stdio.h>
#include<string.h>
const int N = 10005;
const int M = 50005;
int n, m;
struct Edge {
int v1, v2;
int next1, next2;
};
Edge edge[M];
int fir1[N], fir2[N];
int tstamp[N], vis[N];
int scc[N];
int outdeg[N];
int cnt, p;
void init() {
cnt = p = 0;
memset(fir1, -1, sizeof(fir1));
memset(fir2, -1, sizeof(fir2));
memset(outdeg, 0, sizeof(outdeg));
}
void addEdge(int u, int v) {
edge[cnt].v1 = v;
edge[cnt].next1 = fir1[u];
fir1[u] = cnt;
edge[cnt].v2 = u;
edge[cnt].next2 = fir2[v];
fir2[v] = cnt++;
}
void dfs1(int u) {
vis[u] = 1;
for(int i = fir1[u]; i != -1; i = edge[i].next1) {
int v = edge[i].v1;
if(!vis[v])
dfs1(v);
}
tstamp[p++] = u;
}
void dfs2(int u, int st) {
vis[u] = 1;
scc[u] = st;
for(int i = fir2[u]; i != -1; i = edge[i].next2) {
int v = edge[i].v2;
if(!vis[v])
dfs2(v, st);
}
}
int kosaraju() {
memset(vis, 0, sizeof(vis));
for(int i = 1; i <= n; i++)
if(!vis[i])
dfs1(i);
memset(vis, 0, sizeof(vis));
int sccNum = 1;
for(int i = n-1; i >= 0; i--) {
if(!vis[tstamp[i]])
dfs2(tstamp[i], sccNum++);
}
return sccNum;
}
void solve() {
int sccNum = kosaraju();
int ans = 0;
for(int i = 1; i <= n; i++)
for(int j = fir1[i]; j != -1; j = edge[j].next1) {
int v = edge[j].v1;
if(scc[i] != scc[v]) {
outdeg[scc[i]]++;
}
}
int tmp = 0;
int mark;
for(int i = 1; i < sccNum; i++)
if(outdeg[i] == 0){
tmp++;
mark = i;
}
if(tmp == 0)
printf("%d\n", n);
else if(tmp > 1)
printf("0\n");
else {
for(int i = 1; i <= n; i++)
if(scc[i] == mark)
ans++;
printf("%d\n", ans);
}
}
int main () {
while(~scanf("%d%d", &n, &m)) {
int u, v;
init();
for(int i = 0; i < m; i++) {
scanf("%d%d", &u, &v);
addEdge(u, v);
}
solve();
}
return 0;
}