Acwing 1137. 选择最佳线路【图论、单源最短路】

有一天，琪琪想乘坐公交车去拜访她的一位朋友。

由于琪琪非常容易晕车，所以她想尽快到达朋友家。

现在给定你一张城市交通路线图，上面包含城市的公交站台以及公交线路的具体分布。

已知城市中共包含 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];
    }
}