LeetCode解题分享：743. Network Delay Time

Problem

There are N network nodes, labelled 1 to N.

Given times, a list of travel times as directed edges times[i] = (u, v, w), where u is the source node, v is the target node, and w is the time it takes for a signal to travel from source to target.

Now, we send a signal from a certain node K. How long will it take for all nodes to receive the signal? If it is impossible, return -1.

Example 1:

Input: times = [[2,1,1],[2,3,1],[3,4,1]], N = 4, K = 2
Output: 2

Note:

1. N will be in the range [1, 100].
2. K will be in the range [1, N].
3. The length of times will be in the range [1, 6000].
4. All edges times[i] = (u, v, w) will have 1 <= u, v <= N and 0 <= w <= 100.

解题思路

   经典的Dijkstra算法例题。

   代码如下：

class Solution {
public:
    int networkDelayTime(vector<vector<int>>& times, int N, int K) {
        const int inf = INT_MAX;

		vector<vector<int>> graph(N, vector<int>(N, inf));

		vector<bool> visited(N, false);

		vector<int> dist(N, inf);

		dist[K - 1] = 0;

		for (auto& edge : times)
		{
			graph[edge[0] - 1][edge[1] - 1] = edge[2];
		}

		for (int i = 0; i < N; ++i)
		{
			graph[i][i] = 0;
		}

		for (auto loop = 0; loop < N; ++loop)
		{
			int pos = -1;
			int dis = inf;
			for (auto i = 0; i < N; ++i)
			{
				if (visited[i] == false && dist[i] < dis)
				{
					dis = dist[i];
					pos = i;
				}
			}

			if (pos != -1)
			{
				visited[pos] = true;
				for (int i = 0; i < N; ++i)
				{
					if (graph[pos][i] != inf)
					{
						if (dist[i] > dist[pos] + graph[pos][i])
						{
							dist[i] = dist[pos] + graph[pos][i];
						}
					}
				}
			}
		}
		int ans = *max_element(dist.begin(), dist.end());
		if (ans == inf)
		{
			return -1;
		}
		else
		{
			return ans;
		}
		// return 0;
    }
};