[M最短路] lc743. 网络延迟时间(最短路+spfa)

Posted 2021-08-30 Ypuyu

文章目录

• • 1. 题目来源
  • 2. 题目解析

1. 题目来源

链接：743. 网络延迟时间

前置知识：[图+最短路+模板] 五大最短路常用模板

2. 题目解析

有向图单源最短路，数据范围小，啥都能做。没啥好讲的，用的 spfa。

---

• 时间复杂度：$O(m)$
• 空间复杂度：$O(n)$

---

代码：

默写板子就行了，裸题。

const int N = 105, M = 6005;
int h[N], w[M], e[M], ne[M], idx;
int dist[N];
bool st[N];

class Solution {
public:
    void add(int a, int b, int c) {
        e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
    }

    int spfa(int n, int k) {
        memset(dist, 0x3f, sizeof dist);
        queue<int> q;
        q.push(k), dist[k] = 0, st[k] = true;

        while (q.size()) {
            auto t = q.front(); q.pop();
            st[t] = false;

            for (int i = h[t]; ~i; i = ne[i]) {
                int j = e[i];
                if (dist[j] > dist[t] + w[i]) {
                    dist[j] = dist[t] + w[i];
                    if (!st[j]) {
                        st[j] = true;
                        q.push(j);
                    }
                }
            }
        }

        int res = -1e9;
        for (int i = 1; i <= n; i ++ ) res = max(res, dist[i]);
        if (res == 0x3f3f3f3f) res = -1;

        return res;
    }

    int networkDelayTime(vector<vector<int>>& times, int n, int k) {
        memset(h, -1, sizeof h); idx = 0;
        for (auto time : times) add(time[0], time[1], time[2]);
        return spfa(n, k);
    }
};