功能:求单源最短路径(有向图或者无向图)
思路:假设源点为k,维护k到所有节点的最短距离dist[]数组,以及已经找到最短路径的节点i的集合set。
1. 每次寻找不在set中的节点i,且dist[i]最小的节点i
2. 更新源点k通过i能到达的所有节点的最短路径dist
3. 将i存入set中
4. 循环上述操作直到set中包含所有节点
此处以LeetCode-743题为例,java实现代码如下:
public int networkDelayTime(int[][] times, int n, int k) {
//graph[i]中的i表示起点,graph[i]存储i到某一节点的距离
List<int[]>[] graph = new ArrayList[n+1];
for (int i = 1; i < n+1; i++) {
graph[i]=new ArrayList<>();
}
for (int i = 0; i < times.length; i++) {
int begin=times[i][0];
int end=times[i][1];
int value=times[i][2];
graph[begin].add(new int[]{end,value});
}
int[] dist = new int[n+1]; //k点到每个点的距离
for (int i = 1; i < n+1; i++) {
dist[i]=Integer.MAX_VALUE; //初始化所有距离为无穷大
}
dist[k]=0;//k到自己的距离为0
HashSet<Integer> set = new HashSet<>();//存储已找到最短路径的节点
int cur,minDist;
while (set.size()<n){
//寻找距离最小的点cur
cur=-1;
minDist=Integer.MAX_VALUE;
for (int i = 1; i < n+1; i++) {
if (minDist>dist[i]&&!set.contains(i)){
cur=i;
minDist=dist[i];
}
}
if (cur==-1) return -1; //已经没有最小距离的点,此时存在点不可达,退出
//将cur加入set
set.add(cur);
//更新源节点k通过cur走到其他节点的最短路径
for (int i = 0; i < graph[cur].size(); i++) {
int[] temp = graph[cur].get(i);
int end=temp[0];
int value=temp[1];
dist[end]=Math.min(dist[end],dist[cur]+value);
}
}
int res=0;
for (int i = 1; i < n+1; i++) {
res=Math.max(res,dist[i]);
}
return res==Integer.MAX_VALUE?-1:res;
}