SPFA单源最短路算法

SPFA单源最短路算法

适用情况

与 Dijkstra 算法类似，只不过该算法可以判断是否存在负环（不能处理负环）。目的是求单源最短路。

时间复杂度为 O(nm)

该算法可由 Dijkstra 算法替换使用，优于其在于可判断负环。

思想

Bellman-Ford 的改进版。（下述代码采用领接表存储数据）

①【初始化（队列、标记数组等）】创建一个队列，将源点加入队列中并做相关处理（加标记、统计入队次数），然后循环直到队列为空。

②【遍历相邻点并更新数据】循环中，每次取队首顶点（做标记），然后遍历与之相邻的所有点。比较 此队首顶点到源点的距离加上其相邻点之间的距离（即为队首顶点到源点距离+队首顶点到与之相邻点的距离和） 和 相邻点到源点的距离（即源点到与之相邻点的原始值）大小，更新相邻点到源点的距离使之最小。

③【入队并判负环】找到可以更新后，如果没有标记此点则标记再入队，并统计入队次数（如果入队次数大于等于 n 时则表示存在负环）

---

import java.util.Arrays;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class Main {

	static class edge{
		int to,v,nex;
	}
	static int M = 200005;
	static int N = 20005;
	static int cnt = 0;
	static edge[] e = new edge[M];
	
	static int n, m;
	static int[] h = new int[N];//父节点集
	static int[] dis = new int[N];//源点到i的距离
	static boolean[] f = new boolean[N];//标记是否入队
	static int[] k = new int[N];//入队次数
	
	public static void main(String[] args) {
		Scanner cin = new Scanner(System.in);
		
		n = cin.nextInt();//顶点数
		m = cin.nextInt();//边数
		// a 到 b 的距离为 c
		for(int i=0;i<m;i++) {
			int a = cin.nextInt();
			int b = cin.nextInt();
			int c = cin.nextInt();
			add(a,b,c);//此为有向图
		}
		SPFA(1);
		for(int i=2;i<=n;i++) {
			System.out.println(dis[i]);
		}
	}

	private static void SPFA(int s) {
		Queue<Integer> q = new LinkedList<>();
		Arrays.fill(dis, Integer.MAX_VALUE);
		Arrays.fill(f, false);
		Arrays.fill(k, 0);
		
		q.add(s);
		dis[s] = 0;
		f[s] = true;
		while(!q.isEmpty()) {
			int t = q.poll();
			f[t] = false;
			
			for(int i=h[t]; ;i=e[i].nex) {
				if(e[i] == null)break;
				
				int temp = e[i].to;
				if(dis[temp] > dis[t] + e[i].v) {
					dis[temp] = dis[t] + e[i].v;
					
					if(f[temp] == false) {
						q.add(temp);
						f[temp] = true;
						if(++k[temp] >= n) {//判断是否存在负环
							System.out.println("存在负环");
							return ;
						}
					}
				}
			}
		}
	}

	private static void add(int a, int b, int c) {
		e[++cnt] = new edge();// cnt从1开始（点编号）
		e[cnt].to = b;
		e[cnt].v = c;
		e[cnt].nex = h[a];
		h[a] = cnt;
	}
}

---

如有错误或不合理的地方，敬请指正~

加油！！