LeetCode 743.网络延迟时间

LeetCode 743.网络延迟时间

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

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

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

样例输入：
[[2,1,1],[2,3,1],[3,4,1]]//道路信息
4						 //总节点数
2						 //出发的结点

样例输出：
2						 //最长时间

就是单源最短路找时间最长的那个

class Solution
{
  private:
    static const int maxn = 105;
    static const int emax = 6005;
    static const int inf = 0x3f3f3f3f;
    int dist[maxn], tot = 0;
    int head[emax], Next[emax], ver[emax], edge[emax];
    bool vis[maxn];

    void add(int x, int y, int z)//邻接表
    {
        Next[++tot] = head[x], head[x] = tot;
        ver[tot] = y, edge[tot] = z;
    }

    void dijkstar(int st)
    {
        dist[st] = 0;
        priority_queue<pair<int, int>> q;
        q.push(make_pair(-dist[st], st));

        while (!q.empty())
        {
            int x = q.top().second;
            q.pop();
            if (vis[x])
                continue;
            vis[x] = 1;

            for (int i = head[x]; i; i = Next[i])
            {
                int y = ver[i], z = edge[i];
                if (dist[y] > dist[x] + z)
                {
                    dist[y] = dist[x] + z;
                    q.push(make_pair(-dist[y], y));
                }
            }
        }
    }

  public:
    int networkDelayTime(vector<vector<int>> &times, int N, int K)
    {
        memset(vis, 0, sizeof(vis));
        memset(dist, 0x3f, sizeof(dist));
        
        for (int i = 0; i < times.size(); ++i)
            add(times[i][0], times[i][1], times[i][2]);
        
        dijkstar(K);
        int ans = *max_element(dist + 1, dist + N + 1);
        return ans == inf ? -1 : ans;
    }
};