前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 815. 公交路线(最少换乘,BFS)

LeetCode 815. 公交路线(最少换乘,BFS)

作者头像
Michael阿明
发布2020-07-13 15:08:05
1.1K0
发布2020-07-13 15:08:05
举报
文章被收录于专栏:Michael阿明学习之路

1. 题目

我们有一系列公交路线。每一条路线 routes[i] 上都有一辆公交车在上面循环行驶。 例如,有一条路线 routes[0] = [1, 5, 7],表示第一辆 (下标为0) 公交车会一直按照 1->5->7->1->5->7->1->… 的车站路线行驶。

假设我们从 S 车站开始(初始时不在公交车上),要去往 T 站。 期间仅可乘坐公交车,求出最少乘坐的公交车数量。 返回 -1 表示不可能到达终点车站。

代码语言:javascript
复制
示例:
输入: 
routes = [[1, 2, 7], [3, 6, 7]]
S = 1
T = 6
输出: 2
解释: 
最优策略是先乘坐第一辆公交车到达车站 7, 
然后换乘第二辆公交车到车站 6。

说明:
1 <= routes.length <= 500.
1 <= routes[i].length <= 500.
0 <= routes[i][j] < 10 ^ 6.

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/bus-routes 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

代码语言:javascript
复制
class Solution {
public:
    int numBusesToDestination(vector<vector<int>>& routes, int S, int T) {
        if(S == T)//哈哈,又被坑了
            return 0;
        unordered_map<int,unordered_set<int>> station_line;//站点,线路
        unordered_map<int,unordered_set<int>> line_station;//线路,站点
        queue<int> q;//线路队列
        int n = routes.size(), line, station, step = 1, size;
        unordered_set<int> visitedStation;
        unordered_set<int> visitedLine;
        for(int i = 0, j; i < n; ++i)
        {
        	for(j = 0; j < routes[i].size(); ++j)
        	{
        		station_line[routes[i][j]].insert(i);
        		line_station[i].insert(routes[i][j]);
        		if(routes[i][j] == S)
        		{
        			q.push(i);//起点站的线路,都放进队列
        			visitedStation.insert(S);
        			visitedLine.insert(i);
        		}
        	}
        }
        while(!q.empty())
        {
        	size = q.size();
        	while(size--)
        	{
        		line = q.front();
        		q.pop();
        		if(line_station[line].count(T))
        			return step;//线路包含终点,返回
        		for(auto st = line_station[line].begin(); st != line_station[line].end(); ++st)
        		{	//查找这条线路经过的所有站
        			if(!visitedStation.count(*st))
        			{	//没访问过的站
        				visitedStation.insert(*st);
        				for(auto l = station_line[*st].begin(); l != station_line[*st].end(); ++l)
        				{	//这个站的所有未访问线路
        					if(!visitedLine.count(*l))
        					{
        						q.push(*l);
        						visitedLine.insert(*l);
        					}
        				}
        			}
        		}
        	}
        	step++;
        }
        return -1;
    }	
};

444 ms 59.5 MB

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2020/05/29 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档