当前位置:网站首页>leetcode743. 网络延迟时间(中等, dijkstra)

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

2022-06-13 01:29:00 重you小垃

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
思路:单源最短路径->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;
    }
};
原网站

版权声明
本文为[重you小垃]所创,转载请带上原文链接,感谢
https://blog.csdn.net/zhangjiaji111/article/details/125250945