POJ2394 Checking an Alibi【Dijkstra】

题目链接：

http://poj.org/problem?id=2394

题目大意：

有F个农场，标号为1~F。1号农场为谷仓。有C头奶牛在这些农场上，他们喜欢偷吃谷子，但是

在偷吃谷子前M秒，有个摄像头会将它们的位置记录下来。各个农场之间有P条通路，并且知道

这P条路的行走世间。如果奶牛可以在M秒之内赶到谷仓，就可以偷吃到谷子。问：有多少奶牛

可以偷吃到谷子，并升序输出它们的序号。

思路：

最短路径问题。用Dijkstra算法求出1号谷仓到各个农场的最短时间，然后看奶牛所在的农场到谷

仓的时间是否小于等于M，如果小于等于则存入结果数组ans[]中。

AC代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int INF = 0xffffff0;
const int MAXN = 510;

int Map[MAXN][MAXN],Dist[MAXN],ans[MAXN];
bool vis[MAXN];

void Dijkstra(int N,int s)
{
for(int i = 1; i <= N; ++i)
if(i != s)
Dist[i] = Map[s][i];

Dist[s] = 0;
vis[s] = 1;
for(int i = 1; i <= N-1; ++i)
{
int Min = INF;
int k = 0;
for(int j = 1; j <= N; ++j)
{
if(!vis[j] && Dist[j] < Min)
{
Min = Dist[j];
k = j;
}
}
if(k == 0)
return ;
vis[k] = 1;
for(int j = 1; j <= N; ++j)
{
if(!vis[j] && Map[k][j] != INF && Dist[j] > Dist[k]+Map[k][j])
Dist[j] = Dist[k] + Map[k][j];
}
}
}

int main()
{
int F,P,C,M,u,v,w,j;
while(cin >> F >> P >> C >> M)
{
for(int i = 1; i <= F; ++i)
for(int j = 1; j <= F; ++j)
Map[i][j] = INF;
for(int i = 1; i <= P; ++i)
{
cin >> u >> v >> w;
if(Map[u][v] > w)
Map[u][v] = Map[v][u] = w;
}
Dijkstra(F,1);
int cnt = 0;
for(int i = 1; i <= C; ++i)
{
cin >> j;
if(Dist[j] <= M)
ans[cnt++] = i;
}

cout << cnt << endl;
for(int i = 0; i < cnt; ++i)
cout << ans[i] << endl;
}

return 0;
}