题目地址:http://poj.org/problem?id=3256
题目求的是,所有奶牛都可以到的草坪有几块
那只要利用SPFA,枚举所有草坪,求出所有奶牛草坪能否到达该草坪就好了
因为要求的是其他草坪能否到达该草坪,所以把边都反向就好了
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int INF=(1<<30);
const int maxn=10000+5;
struct Edge{
int to; int weight;
Edge(int t,int w):to(t),weight(w){}
};
vector<vector<Edge> > G(maxn);
bool inq[maxn]; //判断是否在队列里
int dist[maxn];
int s[maxn];
void Spfa(int s,int n)
{
memset(inq,false,sizeof(inq));
for(int i=0;i<n+1;i++) dist[i]=INF;
queue<int> Q;
dist[s]=0;
inq[s]=true;
Q.push(s);
while(!Q.empty())
{
int u=Q.front(); Q.pop();
inq[u]=false;
for(int i=0;i<G[u].size();i++)
{
int v=G[u][i].to, w=G[u][i].weight;
if(dist[v]>dist[u]+w)
{
dist[v]=dist[u]+w;
if(!inq[v]) Q.push(v),inq[v]=true;
}
}
}
}
int main()
{
int K,N,M,u,v;
cin>>K>>N>>M;
for(int i=0;i<K;i++)
cin>>s[i];
for(int i=0;i<M;i++)
{
cin>>u>>v;
G[v].push_back(Edge(u,1));
}
int cnt=0,ok;
for(int i=1;i<=N;i++)
{
Spfa(i,N);
ok=1;
for(int j=0;j<K;j++)
if(dist[s[j]]==INF) {
ok=0;
break;
}
if(ok) cnt++;
}
cout<<cnt<<endl;
return 0;
}