题目大意:
有 n 头牛,m 个关系,每个关系对应一个a 和 b ,代表 a 喜欢 b ,如果 a 喜欢 b ,b 喜欢 c ,那么 a 也喜欢 c ,(关系可传递),但是 a 喜欢 b 不代表 b 喜欢 a ,求有几头牛是受所有牛喜欢的
思路:
将 a 喜欢 b 这样的关系抽象成一条 a 指向 b 的有向边,可以建立一个有向图,求有向图中的强连通分量(因为同一个强连通分量内的点可以互相到达,即互相喜欢),然后缩点,将每一个强连通分量缩成一个点,再次建图,对于第二次建的而言,如果图中恰好只有一个点的出度为0,说明该点对应的强连通分量内点的个数就是解,如果有多于一个点出度为0,说明他们之间不能全部连通,即无法达成让任何一头牛得到其他所有牛的喜欢。
第一步,tarjan 求强连通分量
void tarjan(int x){
v[x] = 1;
st[++top] = x;
dfn[x] = low[x] = ++num;
for (int i = head[x]; i; i = nex[i]){
int y = node[i];
if(!dfn[y]){
tarjan(y);
low[x] = min(low[x], low[y]);
}
else if(v[y]){
low[x] = min(low[x], dfn[y]);
}
}
if(dfn[x] == low[x]){
++cnt;
while (1){
int t = st[top--];
v[t] = 0;
c[t] = cnt;
Count[cnt]++;
if(t == x)break;
}
}
}
第二步,缩点(同一个强连通分量内的点可缩成一个点),用一个数组c[ ]记录每一个点所在的强连通分量,用另一个数组 Count[ ]记录对应的强连通分量的大小,(Count[ i ] 表示第 i 个强连通分量的大小)
第三步,重新建图(遍历原始的图的每一个顶点,然后依次遍历邻接边,如果出现两个点所在的强连通分量不相等,则建立一条由顶点的强连通分量指向邻接边对应点的强连通分量的有向边)
for (int i = 1; i <= n; i++){
for (int j = head[i]; j; j = nex[j]){
if(c[i] != c[node[j]]){// c[i]记录的是点i所在的强连通分量
add1(c[i], c[node[j]]);
}
}
}
第四步,遍历新图,判断有多少个点的出度为0(即没有邻接边),并记录出度为0的点所在强连通分量的大小
int ans = 0, res = 0;
for (int i = 1; i <= cnt; i++){
if(head1[i] == 0){
++res;
ans = Count[i];
}
}
完整代码
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn = 50010;
int n, m, k, top, num, cnt, low[maxn], dfn[maxn], node[maxn], nex[maxn], head[maxn], v[maxn], st[maxn], Count[maxn], c[maxn];
int node1[maxn], head1[maxn], nex1[maxn];
void add(int x, int y){
++k;
node[k] = y;
nex[k] = head[x];
head[x] = k;
}
void add1(int x, int y){
++k;
node1[k] = y;
nex1[k] = head1[x];
head1[x] = k;
}
void init(){
cnt = 0;
num = 0;
k = 0;
top = 0;
scanf("%d %d", &n, &m);
for (int i = 1; i <= m; i++){
int a, b;
scanf("%d %d", &a, &b);
add(a, b);
}
}
void tarjan(int x){
v[x] = 1;
st[++top] = x;
dfn[x] = low[x] = ++num;
for (int i = head[x]; i; i = nex[i]){
int y = node[i];
if(!dfn[y]){
tarjan(y);
low[x] = min(low[x], low[y]);
}
else if(v[y]){
low[x] = min(low[x], dfn[y]);
}
}
if(dfn[x] == low[x]){
++cnt;
while (1){
int t = st[top--];
v[t] = 0;
c[t] = cnt;
Count[cnt]++;
if(t == x)break;
}
}
}
void solve(){
for (int i = 1; i <= n; i++){
if(!dfn[i])tarjan(i);
}
k = 0;
for (int i = 1; i <= n; i++){
for (int j = head[i]; j; j = nex[j]){
if(c[i] != c[node[j]]){
add1(c[i], c[node[j]]);
}
}
}
int ans = 0, res = 0;
for (int i = 1; i <= cnt; i++){
if(head1[i] == 0){
++res;
ans = Count[i];
}
}
if(res == 1)printf("%d\n", ans);
else puts("0");
}
int main()
{
init();
solve();
return 0;
}