题目大意:
输入n,m,s 分别代表公交站的数量(1~n),接下来输入m条路径, s是kiki朋友家的公交站。
最后输入一个数字w,代表kiki可以从下面w行数字中的任意站点上车。
求kiki到他朋友家的最短路径。如果不存在则输出-1。
简单分析:
由于起点存在多个,终点仅仅存在一个。那么可以倒着做这道题,用Dijkstra即可。
注意:
1、站点是之间的路是有向图,之前当成无向图WA了一次。
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <algorithm>
#include <iostream>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <assert.h>
#include <time.h>
typedef long long LL;
const int INF = 500000001;
const double EPS = 1e-9;
const double PI = acos(-1.0);
using namespace std;
#define Max 1001
int n, m, s;
int graph[Max][Max], vis[Max], dis[Max];
void init()
{
for(int i = 1; i <= n; i++)
{
for(int j = 1; j <= n; j++)
{
if(i == j) graph[i][j] = 0;
else graph[i][j] = INF;
}
}
}
void Dijkstra()
{
for(int i = 1; i <= n; i++)
{
dis[i] = graph[s][i];
vis[i] = -1;
}
vis[s] = 1;
for(int i = 1; i < n; i++)
{
int minn = INF;
int k = -1;
for(int j = 1; j <= n; j++)
{
if(vis[j] == -1 && minn > dis[j])
{
minn = dis[j];
k = j;
}
}
if(k == -1) break;
vis[k] = 1;
for(int j = 1; j <= n; j++)
{
if(vis[j] == -1 && dis[k] + graph[k][j] < dis[j])
{
dis[j] = dis[k] + graph[k][j];
}
}
}
}
int main()
{
#ifdef _1Test
freopen("test0.in", "r", stdin);
freopen("test0.out", "w", stdout);
srand(time(NULL));
#endif
int a, b, v, w, k;
while(~scanf("%d %d %d", &n, &m, &s))
{
init();
for(int i = 0; i < m; i++)
{
scanf("%d %d %d", &a, &b, &v);
graph[b][a] = min(v, graph[b][a]);
}
scanf("%d", &w);
int ans = INF;
Dijkstra();
while(w--)
{
scanf("%d", &k);
ans = min(ans, dis[k]);
}
if(ans == INF) puts("-1");
else printf("%d\n", ans);
}
return 0;
}