Leetcode 743：网络延迟时间

题目描述

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

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

现在，我们向当前的节点 K 发送了一个信号。需要多久才能使所有节点都收到信号？如果不能使所有节点收到信号，返回 -1。

注意:

N 的范围在 [1, 100] 之间。
K 的范围在 [1, N] 之间。
times 的长度在 [1, 6000] 之间。
所有的边 times[i] = (u, v, w) 都有 1 <= u, v <= N 且 0 <= w <= 100。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/network-delay-time
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

解题思路

最短路径

class Solution {
public:
    void dijkstra(vector<vector<pair<int,int>>>& graph,vector<int>& vis,vector<int>& dis,int N,int k){
        dis[k] = 0;
        for(int i=1;i<=N;i++){
            int u = -1,tmp = INT_MAX;
            for(int i=1;i<=N;i++){
                if(vis[i]==0 && dis[i] < tmp){
                    u = i;
                    tmp = dis[i];
                }
            }
            if(u == -1) return;
            vis[u] = 1;
            for(int i=0;i<graph[u].size();i++){
                int v = graph[u][i].first;
                if(vis[v] == 0 && dis[u]+graph[u][i].second < dis[v]){
                    dis[v] = dis[u] + graph[u][i].second;
                }
            }
        }
    }
    int networkDelayTime(vector<vector<int>>& times, int N, int K) {
        int ans = 0;
        if(times.size() == 0||times[0].size() == 0) return 0;
        int row = times.size(),col = times[0].size();
        vector<vector<pair<int,int>>> graph(N+1);
        vector<int> vis(N+1,0);
        vector<int> dis(N+1,INT_MAX);
        for(int i=0;i<row;i++){
            for(int j=0;j<col;j++) graph[times[i][0]].push_back(make_pair(times[i][1],times[i][2]));
        }
        //dijkstra
        dijkstra(graph,vis,dis,N,K);
        for(int i=1;i<=N;i++){
            if(vis[i] == 0) return -1;
        }
        for(int i=1;i<=N;i++) ans = max(ans,dis[i]);
        return ans;
    }
};