LeetCode 743. 网络延迟时间

Posted 2021-10-12 Blocking The Sky

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

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

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

迪杰斯特拉算法

class Solution {
public:
    int networkDelayTime(vector<vector<int>>& times, int n, int k) {//迪杰斯特拉算法
        bool set[n+1];//标记数组，set[vi]=1表示vi并入最短路径
        int dist[n+1];//最短路径长度
        int edge[n+1][n+1];
        int INTMAX=10000;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i==j){
                    edge[i][j]=0;
                }
                else 
                    edge[i][j]=INTMAX;
            }
        }
        for(int i=0;i<times.size();i++){
            edge[times[i][0]][times[i][1]]=times[i][2];
        }
        int minn;
        for(int i=1;i<=n;i++){
            dist[i]=edge[k][i];
            set[i]=false;
        }
        set[k]=true;
        /*初试化结束*/
        /*关键操作开始*/
        int u;
        for(int i=0;i<n-1;i++){
            minn=INT_MAX;
            /*这个循环每次从剩余顶点中选出一个顶点，通往这些顶点的路径在通往所有剩余顶点的路径中长度是最短的*/
            for(int j=1;j<=n;j++){
                if(!set[j]&&minn>dist[j]){
                    minn=dist[j];
                    u=j;      
                }
            }
            set[u]=true;//将选出的顶点并入最短路径中
            /*这个循环以刚并入的顶点作为中间点，对所有通往剩余顶点的路径进行检测*/
            for(int j=1;j<=n;j++){
                /*判断顶点u的加入是否会出现通往顶点j的最短路径,如果出现则更新，否则什么也不做*/
                if(!set[j]&&(dist[u]+edge[u][j]<dist[j])){
                    dist[j]=dist[u]+edge[u][j];
                }
            }
        }
        int maxx=INT_MIN;
        for(int i=1;i<=n;i++){
            if(maxx<dist[i])
                maxx=dist[i];
        }
        if(maxx<INTMAX)
            return maxx;
        else 
            return -1;
    }
};