743. 网络延迟时间(dijkstral算法)

743. 网络延迟时间

有 n 个网络节点，标记为 1 到 n。

给你一个列表 times，表示信号经过 有向 边的传递时间。 times[i] = (ui, vi, wi)，其中 ui 是源节点，vi 是目标节点， wi 是一个信号从源节点传递到目标节点的时间。

现在，从某个节点 K 发出一个信号。需要多久才能使所有节点都收到信号？如果不能使所有节点收到信号，返回 -1 。

 

示例 1：

输入：times = [[2,1,1],[2,3,1],[3,4,1]], n = 4, k = 2
输出：2

示例 2：

输入：times = [[1,2,1]], n = 2, k = 1
输出：1

示例 3：

输入：times = [[1,2,1]], n = 2, k = 2
输出：-1

 

提示：

• 1 <= k <= n <= 100
• 1 <= times.length <= 6000
• times[i].length == 3
• 1 <= ui, vi <= n
• ui != vi
• 0 <= wi <= 100
• 所有 (ui, vi) 对都 互不相同（即，不含重复边）

class Solution {
public:
    vector<int> dijkstral(const vector<vector<int>> &graph, int start) {
        int n = graph.size();
        vector<int> dis(n, INF); // 初始化起点到各节点最短距离为无穷大
        vector<bool> visited(n, false); // 初始化所有节点未被选过
        dis[start] = 0; // 初始起点到起点距离为0
        for (int i = 0; i < n - 1; i++) {
            int minNode = -1;
            // 选定一个节点:
            // 1)未被选过;
            // 2)距离最短;
            for (int j = 0; j < n; j++) {
                if (!visited[j] && ((minNode == -1) || (dis[j] < dis[minNode]))) {
                    minNode = j;
                }
            }
            // 对于已选择的节点作为中转站,刷新经过中转站到各节点的最短路径
            for (int j = 0; j < n; j++) {
                dis[j] = std::min(dis[j], dis[minNode] + graph[minNode][j]);
            }
            visited[minNode] = true;
        }
        return dis;
    }
    static constexpr int INF = 0x3F3F3F3F;
    int networkDelayTime(vector<vector<int>>& times, int n, int k) {
        vector<vector<int>> graph(n, vector<int>(n, INF));
        for (auto &vec : times) {            
            graph[vec[0] - 1][vec[1] - 1] = vec[2];
        }
        vector<int> ans;
        ans = dijkstral(graph, k - 1);
        sort(ans.begin(), ans.end());
        int size = ans.size();
        return (ans[size - 1] == INF) ? -1 : ans[size - 1];
    }
};