leetcode743. 网络延迟时间(中等, dijkstra)

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
思路:单源最短路径->dijkstra 模板

class Solution {
public:
    using PII = pair<int, int>;
    int networkDelayTime(vector<vector<int>>& times, int n, int k) {
        
        vector<int> vis(n + 1), dis(n + 1, INT_MAX);
        vector<vector<PII>> bian(n + 1); //目标,距离
        for (auto& a : times) {
            bian[a[0]].push_back({a[1], a[2]});
        }
        priority_queue<PII, vector<PII>, greater<PII>> pq; //距离->最近目的地
        pq.push({0, k});
        int cnt = 0;
        dis[k] = 0;
        while (!pq.empty()) {
            auto [_dis, pos] = pq.top();
            pq.pop();
            if (vis[pos]) continue;
            vis[pos] = true;
            cnt++;
            for (auto& tmp : bian[pos]) {
                if (_dis + tmp.second < dis[tmp.first]) {
                    dis[tmp.first] = _dis + tmp.second;
                    pq.push({dis[tmp.first], tmp.first});
                }
            }
        }
        int ans = 0;
        for (int i = 1; i <= n; ++i) {
            ans = max(ans, dis[i]);
        }
        return (cnt == n) ? ans : -1;
    }
};

版权声明:本文为zhangjiaji111原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。