链接地址:http://poj.org/problem?id=2186
题意:n头奶牛,给出若干个欢迎关系a b,表示a欢迎b,欢迎关系是单向的,但是是可以传递的。另外每个奶牛都是欢迎他自己的。求出被所有的奶牛欢迎的奶牛的数目。
思路:N个顶点的有向图,有M条边(N≤10000,M≤50000)。求一共有多少个点,满足这样的条件:所有其它的点都可以到达这个点。
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <iterator>
const int maxn = 50050;
using namespace std;
int V;
vector <int> g[maxn];
vector <int> rg[maxn];
vector <int> vs;
int use[maxn];
int cmp[maxn];
int N, M;
int A[maxn] , B[maxn];
void add(int f, int t)
{
g[f].push_back(t);
rg[t].push_back(f);
}
void dfs(int v)
{
use[v] = 1;
for(int i = 0; i < g[v].size(); i++)
{
if(!use[g[v][i]])
dfs(g[v][i]);
}
vs.push_back(v);
}
void rdfs(int v, int k)
{
use[v] = 1;
cmp[v] = k;
for(int i = 0; i < rg[v].size(); i++)
if(!use[rg[v][i]])
rdfs(rg[v][i], k);
}
int scc()
{
memset(use, 0, sizeof(use));
vs.clear();
for(int v = 0; v < V; v++)
if(!use[v])
dfs(v);
memset(use, 0, sizeof(use));
int k = 0;
for(int i = vs.size() - 1; i >= 0; i--)
if(!use[vs[i]])
rdfs(vs[i], k++);
return k;
}
int main()
{
cin >> N >> M;
V = N;
for(int i = 0; i < M; i++)
{
scanf("%d%d", &A[i], &B[i]);
add(A[i] - 1, B[i] - 1);
}
int n = scc(); //统计备选解的个数
int u = 0, num = 0;
for(int v = 0; v < V; v++)
{
if(cmp[v] == n - 1)
{
u = v;
num++;
}
}
memset(use, 0, sizeof(use));
rdfs(u, 0);
for(int v = 0; v < V; v++)
{
if(!use[v])
{
num = 0;
break;
}
}
cout << num << endl;
return 0;
}