815. Bus Routes
https://leetcode.com/problems/bus-routes/description/
/*
* @lc app=leetcode id=815 lang=cpp
*
* [815] Bus Routes
*/
// @lc code=start
class Solution {
public:
int numBusesToDestination(vector<vector<int>>& routes, int S, int T) {
if (S == T) return 0;
const int N = routes.size();
unordered_map<int, vector<int>> stops;
for (int i = 0; i < N; ++i) {
for (const auto s : routes[i]) {
stops[s].push_back(i);
}
}
vector<bool> visited(N, false);
queue<int> q;
q.push(S);
int res = 0;
while (!q.empty()) {
++res;
for (int i = q.size() - 1; i >= 0; --i) {
const auto t = q.front(); q.pop();
for (const auto b : stops[t]) {
if (visited[b]) continue;
visited[b] = true;
for (const auto s : routes[b]) {
if (s == T) return res;
q.push(s);
}
}
}
}
return -1;
}
};
// @lc code=end
Last updated