【Java模板】SPFA模板

最新推荐文章于 2024-05-15 16:01:10 发布

原创最新推荐文章于 2024-05-15 16:01:10 发布 · 479 阅读

1 ·

CC 4.0 BY-SA版权

Java模板专栏收录该内容

9 篇文章

订阅专栏

本文详细介绍了一种基于邻接表实现的SPFA算法，用于解决有向图中的最短路径问题。通过实例展示了算法的初始化过程、边的添加及SPFA算法的具体实现。特别关注了如何检测负权环的存在。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

基于邻接表（链表形式）实现。

import java.util.*;
public class SPFA模板 {
	static int N = 1005;
	static int len; //有向边个数 
	static int[] head = new int[N];
	static edge[] e = new edge[N];
	static int[] dis = new int[N];
	static boolean[] inq = new boolean[N]; //判断是否在队列里面
	static int[] cntInq = new  int[N]; //记录一个点入队列的次数 来判断是否有环
	
	
	static void add(int u, int v, int w) {
		e[len] = new edge(v, w, head[u]);
		head[u] = len++;
	}
	static void add2(int u, int v, int w) {
		add(u, v, w);
		add(v, u, w);
	}
	static void init() {
		Arrays.fill(head, -1);
		Arrays.fill(dis, 0x3f3f3f3f);
	}
	
	static int n,m;
	public static void main(String[] args) {
		init();
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		m = sc.nextInt();
		while (m-- > 0) {
			add2(sc.nextInt(), sc.nextInt(), sc.nextInt());
		}
		if (spfa(1)){
			System.out.println("有负环");
		} else {
			System.out.println(dis[n]);
		}
	}
	static boolean spfa(int u) {
		dis[u] = 0;//起点标记为0
		inq[u] = true; //放入队列
		cntInq[u]++;  //入队次数增加
		
		Queue<Integer> q = new LinkedList<Integer>();
		q.add(u);
		
		while (q.size() != 0) {
			u = q.poll();
			inq[u] = false;
			
			for (int j = head[u]; j != -1; j = e[j].next) {
				int v = e[j].v;
				int w = e[j].w;
				if (dis[v] > dis[u] + w) {
					//进行跟新, 如果不在队中还应该放入队中
					dis[v] = dis[u] + w;
					if (!inq[v]) {
						//入队 且 次数加1
						cntInq[v]++;
						q.add(v);
						inq[v] = true;
						
						if (cntInq[v] > n) return true; //表明有负环
					}
				}
			}
		}
		
		return false;
	}
}