牛客网 2018校招真题 美团点评 公交车

Description

牛客网 2018校招真题 公交车

Solving Ideas

建图：
将公交车抽象成一个站点，将公交车与其可达站点的关系作为无向图的边，边集只记录公交车与站点的关系。

求最小花费：
从起点开始使用bfs算法求最短距离，因为设立了抽象点，起点距离又设为1，所以每个点的距离是实际距离的两倍加一，例如

起点为1，n = 5，由bfs得dist[n] = 5，但实际花费为2

Solution

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

/**
 * @author wylu
 */
public class Main {
    static ArrayList<Integer>[] graph;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] strs = br.readLine().split(" ");
        int n = Integer.parseInt(strs[0]), m = Integer.parseInt(strs[1]);
        graph = new ArrayList[n + m + 1];

        //建图, 将公交车抽象成一个站点, 只记录公交车可到站点的边
        for (int i = 1; i <= m; i++) {
            strs = br.readLine().split(" ");
            graph[i + n] = new ArrayList<>();
            for (int j = 1; j < strs.length; j++) {
                int station = Integer.parseInt(strs[j]);
                if (graph[station] == null) graph[station] = new ArrayList<>();
                graph[station].add(i + n);
                graph[i + n].add(station);
            }
        }

        //bfs求最短路, 
        int res = bfs(1, n, m);
        //因为设立了抽象点,起点距离又设为1,所以每个点的距离是实际距离的两倍加一
        System.out.println(res == 0 ? "-1" : res / 2);
    }

    private static int bfs(int x, int n, int m) {
        int[] dist = new int[n + m + 1];
        LinkedList<Integer> queue = new LinkedList<>();
        dist[x] = 1;
        queue.offer(x);
        while (!queue.isEmpty()) {
            int cur = queue.poll();
            for (Integer e : graph[cur]) {
                if (dist[e] == 0) {
                    dist[e] = dist[cur] + 1;
                    queue.offer(e);
                }
            }
        }
        return dist[n];
    }
}