(nice!!!) LeetCode 815. 公交路线（哈希表、广度优先搜索bfs）

最新推荐文章于 2024-10-17 14:59:31 发布

岁忧

最新推荐文章于 2024-10-17 14:59:31 发布

阅读量389

点赞数 6

分类专栏： LeetCode 文章标签： leetcode 散列表宽度优先 c++ 算法广度优先搜索哈希表

本文链接：https://blog.csdn.net/weixin_46028214/article/details/142316449

版权

LeetCode 专栏收录该内容

527 篇文章 0 订阅

订阅专栏

题目：815. 公交路线

在这里插入图片描述

思路：用哈希表mp来存储每个站点能在哪些路线出现，再进行广度优先搜索bfs，每个站点只需遍历一次。时间复杂度为0(10^5)，细节看注释。

class Solution {
public:
    int numBusesToDestination(vector<vector<int>>& routes, int source, int target) {
        if(source==target) return 0;
        //哈希表mp来存储每个站点能从哪些路线出现
        unordered_map<int,vector<int>> mp;
        //遍历所有路线
        for(int i=0;i<routes.size();i++){
        	//遍历路线的所有站点
            for(auto x:routes[i]){
            	//将路线插入到站点的哈希表mp当中
                mp[x].push_back(i);
            }
        }
        //如果没有路线到达起始点\终点
        if(mp[source].size()==0||mp[target].size()==0) return -1;
        //队列，进行广度优先搜索bfs
        queue<int> qu;
        //记录到站点的距离
        unordered_map<int,int> dis;
        //初始化
        dis[source]=0;
        qu.push(source);
        //广度优先搜索bfs
        while(qu.size()){
        	//队首元素
            int x=qu.front();
            qu.pop();
            //遇到终点栈
            if(x==target) break;
            //遍历能从站点x出发的路线
            for(auto a:mp[x]){
            	//遍历路线上的所有站点
                for(auto b:routes[a]){
                	//站点出现过，就没必要在插入了，毕竟找最少
                    if(dis.count(b)) continue;
                    dis[b]=dis[x]+1;
                    qu.push(b);
                }
                //遍历过的路线就置空，毕竟已经遍历过一遍，没必要再多跑一趟
                routes[a].clear();
            }
        }
        return dis.count(target)?dis[target]:-1;
    }
};