Leetcode—815. 公交路线【困难】（unordered_map+queue）

最新推荐文章于 2024-10-14 18:52:56 发布

源代码•宸

最新推荐文章于 2024-10-14 18:52:56 发布

阅读量407

点赞数 12

分类专栏： LeetCode刷题文章标签： leetcode 算法 c++ bfs 经验分享

本文链接：https://blog.csdn.net/qq_44631615/article/details/142311724

版权

LeetCode刷题专栏收录该内容

402 篇文章 5 订阅

订阅专栏

2024每日刷题（163）

Leetcode—815. 公交路线

在这里插入图片描述

bfs实现代码

class Solution {
public:
    int numBusesToDestination(vector<vector<int>>& routes, int source, int target) {
        if(source == target) {
            return 0;
        }
        unordered_map<int, vector<int>> stop2bus;
        int m = routes.size();

        for(int i = 0; i < m; i++) {
            for(const auto stop: routes[i]) {
                stop2bus[stop].push_back(i);
            }
        }
        vector<bool> visited(501, false);
        queue<int> q;
        for(auto &bus: stop2bus[source]) {
            visited[bus] = true;
            q.push(bus);
        }

        int ans = 1;
        while(!q.empty()) {
            int size = q.size();
            while(size--) {
                int bus = q.front();
                q.pop();
                for(auto &stop: routes[bus]) {
                    if(stop == target) {
                        return ans;
                    }
                    for(auto &nextBus: stop2bus[stop]) {
                        if(visited[nextBus] == false) {
                            visited[nextBus] = true;
                            q.push(nextBus);
                        }
                    }
                }
            }
            ans++;
        }
        return -1;
    }
};