题意
现在有 N N N头牛,编号从 1 1 1到 N N N,给你 M M M对整数 ( A , B ) (A,B) (A,B),表示牛 A A A认为牛 B B B受欢迎。这种关系是具有传递性的,如果 A A A认为 B B B受欢迎, B B B认为 C C C受欢迎,那么牛 A A A也认为牛 C C C受欢迎。求出有多少头牛被除自己之外的所有牛认为是受欢迎的。
数据范围
1
≤
N
≤
1
0
4
1 \leq N \leq 10^4
1≤N≤104
1
≤
M
≤
5
∗
1
0
4
1 \leq M \leq 5*10^4
1≤M≤5∗104
思路
这道题将图建立出来,形成一个有向图。但是有向图不好处理,因此考虑缩点。
我们直接考虑缩点后的DAG。显然只有出度为0的点才有可能满足要求,因此我们只关注出度为0的点。
如果出度为0的点不只有一个,那么肯定没有点符合要求了,因为他们之间没办法互相受欢迎。
所以最后结果为出度为0的强连通分量中点的个数。
代码
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
using namespace std;
const int N = 10010, M = 50010;
int n,m;
int h[N], e[M], ne[M], idx;
int scc_cnt, timestamp;
bool in_stk[N];
int id[N], sz[N], dfn[N], low[N];
stack<int> stk;;
int dout[N];
void add(int a,int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
void tarjan(int u)
{
low[u] = dfn[u] = ++ timestamp;
stk.push(u), in_stk[u] = true;
for(int i=h[u];~i;i=ne[i]){
int j = e[i];
if(!dfn[j]){
tarjan(j);
low[u] = min(low[j], low[u]);
}
else if(in_stk[j]) low[u] = min(low[u], dfn[j]);
}
if(dfn[u]==low[u]){
int v;
scc_cnt ++;
do{
v = stk.top();
stk.pop();
in_stk[v] = false;
id[v] = scc_cnt;
sz[scc_cnt] ++;
}while(u!=v);
}
}
int main()
{
cin >> n >> m;
memset(h,-1,sizeof(h));
for(int i=1;i<=m;i++){
int a,b;
cin >> a >> b;
add(a,b);
}
for(int i=1;i<=n;i++){
if(!dfn[i]){
tarjan(i);
}
}
for(int i=1;i<=n;i++){
for(int j=h[i];~j;j=ne[j]){
int k = e[j];
int a = id[i], b = id[k];
if(a!=b) dout[a] ++;
}
}
int zeros = 0, sum = 0;
for(int i=1;i<=scc_cnt;i++){
if(!dout[i]){
zeros ++;
sum += sz[i];
if(zeros>1){
sum = 0;
break;
}
}
}
cout << sum << endl;
return 0;
}