LeetCode #815. 公交路线

815. 公交路线

难度困难144收藏分享切换为英文接收动态反馈

给你一个数组 routes ，表示一系列公交线路，其中每个 routes[i] 表示一条公交线路，第 i 辆公交车将会在上面循环行驶。

• 例如，路线 routes[0] = [1, 5, 7] 表示第 0 辆公交车会一直按序列 1 -> 5 -> 7 -> 1 -> 5 -> 7 -> 1 -> ... 这样的车站路线行驶。

现在从 source 车站出发（初始时不在公交车上），要前往 target 车站。 期间仅可乘坐公交车。

求出 最少乘坐的公交车数量 。如果不可能到达终点车站，返回 -1 。

示例 1：

输入：routes = [[1,2,7],[3,6,7]], source = 1, target = 6
输出：2
解释：最优策略是先乘坐第一辆公交车到达车站 7 , 然后换乘第二辆公交车到车站 6 。 

示例 2：

输入：routes = [[7,12],[4,5,15],[6],[15,19],[9,12,13]], source = 15, target = 12
输出：-1

提示：

• 1 <= routes.length <= 500.
• 1 <= routes[i].length <= 105
• routes[i] 中的所有值 互不相同
• sum(routes[i].length) <= 105
• 0 <= routes[i][j] < 106
• 0 <= source, target < 106

每个车站保存自己的车辆下标，这个处理绝了

class Solution {
public:
    vector<int> vec[1000005];
    typedef pair<int,int> P;
    int vis[1000006] = {0};
    int numBusesToDestination(vector<vector<int>>& routes, int source, int target) {
        for(int i = 0;i < routes.size();i++){
            for(int j = 0;j < routes[i].size();j++){
                vec[routes[i][j]].push_back(i);
            }
        }
        vector<int> LEN(1e6 + 1,INT_MAX);
        queue<P> que;
        que.push({0,source});
        LEN[source] = 0;
        while(que.size() != 0){
            P p = que.front();
            que.pop();
            int len = p.first,x = p.second;
            if(x == target) return LEN[x];
            for(int i = 0;i < vec[x].size();i++){
                int r = vec[x][i];
                if(vis[r] == 1) continue;
                vis[r] = 1;
                for(int j = 0;j < routes[r].size();j++){
                    if(LEN[x] + 1 < LEN[routes[r][j]]){
                        LEN[routes[r][j]] = LEN[x] + 1;
                        que.push({LEN[routes[r][j]],routes[r][j]});
                    }
                }
            }
        }
        return LEN[target] == INT_MAX ? -1 : LEN[target];
    }
};