我们有一系列公交路线。每一条路线 routes[i] 上都有一辆公交车在上面循环行驶。例如,有一条路线 routes[0] = [1, 5, 7],表示第一辆 (下标为0) 公交车会一直按照 1->5->7->1->5->7->1->… 的车站路线行驶。
假设我们从 S 车站开始(初始时不在公交车上),要去往 T 站。 期间仅可乘坐公交车,求出最少乘坐的公交车数量。返回 -1 表示不可能到达终点车站。
示例:
输入:
routes = [[1, 2, 7], [3, 6, 7]]
S = 1
T = 6
输出: 2
解释:
最优策略是先乘坐第一辆公交车到达车站 7, 然后换乘第二辆公交车到车站 6。
我的代码 228ms
class Solution {
public:
int numBusesToDestination(vector<vector<int>>& routes, int S, int T) {
if(S==T) return 0;
unordered_map<int,set<int>> m;
queue<list<int>> que;
vector<int> index(routes.size(),0);
int level=1;
for(int i=0;i<routes.size();i++)
for(int j=0;j<routes[i].size();j++)
{
int p=routes[i][j];
m[p].insert(i);
if(p==T)
index[i]=1;
}
for(int i:m[S])
{
list<int> l{i};
que.push(l);
}
while(!que.empty())
{
list<int>t_l=que.front();
que.pop();
if(index[t_l.back()]==1) return t_l.size();
index[t_l.back()]=-1;
for(int i=0;i<routes[t_l.back()].size();i++)
for(int j:m[routes[t_l.back()][i]])
{
if(index[j]!=-1)
{
list<int>t=t_l;
t.push_back(j);
que.push(t);
}
}
}
return -1;
}
};