公交车， 美团笔试题

原来有n个公交车站，有m条公交线路。对于其中一条线路a->b->c,使得a,b,c三个公交车站距离变成了1（1块钱可以到达）。因此，我们建立无向图（边全值为1）。但是公交线路长度很长（2<=t <=100000），这样对一个公交线路来说，建边复杂度为(10^5 imes 10^5)。为了降低复杂度，每个线路都建立一个虚拟节点，使得公交车站通过虚拟节点相连接，这样就减少了边的数量,这样对一个公交线路来说，建边复杂度为(10^5)。对于对于其中一条线路a->b->c， 建立一个节点x，如果要求a到b的距离，a,b之间并没有直接边相连，而是通过x节点，间接相连（a->x->b)。因此，我们将边的权值设为0.5，a到b的权值就仍然为1。问题还是求1-n的距离。

import java.util.*;
public class Main {
    static int bfs(List<Integer>[] g, int n){
        boolean[] used = new boolean[g.length];
        LinkedList<Integer> q = new LinkedList<>();
        q.offer(1); used[1] = true;
        int distance = 0;
        while(!q.isEmpty()){
            int size = q.size();
            distance ++;
            for(int i=0; i < size; i++) {
                int cur = q.poll();
                for(int j=0; j <g[cur].size(); j++) {
                    int v = g[cur].get(j);
                    if(v == n) return distance/2;
                    if(used[v] == false) {
                        used[v] = true;
                        q.offer(v);
                    }
                }
            }
        }
        return -1;
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt(), m = sc.nextInt();
        List[] g = new ArrayList[n+m+1];
        for(int i=1; i < g.length; i++) g[i] = new ArrayList<Integer>();
        for(int i=1; i <= m; i++) {
            int t = sc.nextInt();
            for(int j=1; j <= t; j++) {
                int v = sc.nextInt();
                g[n+i].add(v);
                g[v].add(n+i);
            }
        }
        int res = bfs(g, n);
        System.out.println(res);
    }
}