有一天,琪琪想乘坐公交车去拜访她的一位朋友。
由于琪琪非常容易晕车,所以她想尽快到达朋友家。
现在给定你一张城市交通路线图,上面包含城市的公交站台以及公交线路的具体分布。
已知城市中共包含 n 个车站(编号1~n)以及 m 条公交线路。
每条公交线路都是 单向的,从一个车站出发直接到达另一个车站,两个车站之间可能存在多条公交线路。
琪琪的朋友住在 s 号车站附近。
琪琪可以在任何车站选择换乘其它公共汽车。
请找出琪琪到达她的朋友家(附近的公交车站)需要花费的最少时间。
输入格式
输入包含多组测试数据。
每组测试数据第一行包含三个整数 n,m,s,分别表示车站数量,公交线路数量以及朋友家附近车站的编号。
接下来 m 行,每行包含三个整数 p,q,t,表示存在一条线路从车站 p 到达车站 q,用时为 t。
接下来一行,包含一个整数 w,表示琪琪家附近共有 w 个车站,她可以在这 w 个车站中选择一个车站作为始发站。
再一行,包含 w 个整数,表示琪琪家附近的 w 个车站的编号。
输出格式
每个测试数据输出一个整数作为结果,表示所需花费的最少时间。
如果无法达到朋友家的车站,则输出 -1。
每个结果占一行。
数据范围
n≤1000,m≤20000,
1≤s≤n,
0<w<n,
0<t≤1000
输入样例:
5 8 5
1 2 2
1 5 3
1 3 4
2 4 7
2 5 6
2 3 5
3 5 1
4 5 1
2
2 3
4 3 4
1 2 3
1 3 4
2 3 2
1
1
输出样例:
1
-1
- 本题问多起点到终点的最短距离,对每个起点暴力做一遍spfa肯定超时,而且还是多组测试数据。因此本题需要建立一个虚拟源点,把虚拟源点到每个起点都连一条边权为1的边,此时从集合的角度问题就转换为单个虚拟源点到终点的最短距离。此时只需要做一遍SPFA即可。
- 注意边数数组要开大一点,因为多加了最多n条边。
import java.io.*;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
public class Main {
static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
static PrintWriter out = new PrintWriter(new BufferedWriter(new OutputStreamWriter(System.out)));
static int n, m, end, idx = 0, INF = 0x3f3f3f3f;
static class Edge {
private int v, w, ne;
public Edge(int v, int w, int ne) {
this.v = v;
this.w = w;
this.ne = ne;
}
}
static Edge[] edges;
static int[] dist, h;
static boolean[] st;
public static void main(String[] args) throws IOException {
while (br.ready()) {
String[] line = br.readLine().split(" ");
n = Integer.parseInt(line[0]);
m = Integer.parseInt(line[1]);
end = Integer.parseInt(line[2]);
edges = new Edge[m + n];
dist = new int[n + 1];
h = new int[n + 1];
st = new boolean[n + 1];
Arrays.fill(h, -1);
idx = 0;
while (m-- > 0) {
line = br.readLine().split(" ");
int u = Integer.parseInt(line[0]), v = Integer.parseInt(line[1]), w = Integer.parseInt(line[2]);
add(u, v, w);
}
int s = Integer.parseInt(br.readLine());
line = br.readLine().split(" ");
for (int i = 0; i < s; i++) {
int start = Integer.parseInt(line[i]);
add(0, start, 0);
}
out.println(spfa());
}
br.close();
out.close();
}
public static void add(int u, int v, int w) {
edges[idx] = new Edge(v, w, h[u]);
h[u] = idx++;
}
public static int spfa() {
Arrays.fill(dist, INF);
Queue<Integer> q = new LinkedList<>();
q.offer(0);
dist[0] = 0;
while (!q.isEmpty()) {
int t = q.poll();
st[t] = false;
for (int i = h[t]; i != -1; i = edges[i].ne) {
int v = edges[i].v, w = edges[i].w;
if (dist[t] + w < dist[v]) {
dist[v] = dist[t] + w;
if (!st[v]) {
q.offer(v);
st[v] = true;
}
}
}
}
return dist[end] == INF ? -1 : dist[end];
}
}