当前位置: 代码迷 >> 综合 >> POJ 3256 Cow Picnic SPFA
  详细解决方案

POJ 3256 Cow Picnic SPFA

热度:46   发布时间:2023-09-23 08:07:36.0

题目地址: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;
}