SPFA模板

代码:

#include<bits/stdc++.h>

#define inf 2147483647
#define maxn 10005
#define maxm 500005

using namespace std;

int n, m, s, num_edge=0;
int dis[maxn], vis[maxn], head[maxm];

struct Edge {
	int next, to, dis;
}edge[maxm]; //结构体表示静态邻接表

void addedge( int from, int to, int dis ) {//邻接表建图 以下是数据结构书上的标准代码,不懂翻书看解释
	edge[++num_edge].next = head[from]; //链式存储下一条出边
	edge[num_edge].to = to; //当前节点编号
	edge[num_edge].dis = dis; //本条边的距离
	head[from] = num_edge; //记录下一次的出边情况
}
void spfa() {
	queue<int> q; //spfa用队列,这里用了STL的标准队列
	for(int i=1; i<=n; i++) {
		dis[i] = inf; //带权图初始化
		vis[i] = 0; //记录点i是否在队列中,同dijkstra算法中的visited数组
  	}
	q.push( s ); dis[s] = 0; vis[s] = 1; //第一个顶点入队,进行标记
  	while( !q.empty() ) {
    		int u = q.front(); //取出队首
    		q.pop(); vis[u] = 0; //出队标记
    		for( int i = head[u]; i; i = edge[i].next ) { //邻接表遍历,不多解释了(也可用vector代替)
      			int v = edge[i].to; 
      			if( dis[v] > dis[u] + edge[i].dis ) { //如果有最短路就更改
        			dis[v] = dis[u] + edge[i].dis;
        			if( vis[v] == 0 ) { //未入队则入队
          				vis[v] = 1; //标记入队
         	 			q.push(v);
        			}
      			}
    		}
  	}
}
int main() {
	scanf( "%d%d%d", &n, &m, &s );
  	for( int i = 1; i <= m; i++ ) {
    		int f, g, w;
    		scanf( "%d%d%d", &f, &g, &w ); 
    		addedge( f, g, w ); //建图,有向图连一次边就可以了
  	}
  	spfa(); //开始跑spfa
  	for( int i = 1; i <= n; i++ )
    		if( s == i ) printf( "0 " ); //如果是回到自己,直接输出0
      	else	printf( "%d ", dis[i] ); //否则打印最短距离
  	return 0;
} //结束

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,以下是一个简单的SPFA模板: ``` const int INF = 0x3f3f3f3f; const int MAXN = 1005; vector<pair<int, int>> g[MAXN]; // 存储有向图,pair 中第一个数表示终点,第二个数表示边权 int dist[MAXN]; // 存储源点到每个点的最短距离 bool in_queue[MAXN]; // 标记每个点是否在队列中 void spfa(int s) { // s 表示源点编号 memset(dist, INF, sizeof(dist)); // 初始化 dist 数组为 INF,表示距离无限远 memset(in_queue, 0, sizeof(in_queue)); // 初始化 in_queue 数组为 false,表示所有点都不在队列中 queue<int> q; q.push(s); dist[s] = 0; in_queue[s] = true; while (!q.empty()) { int u = q.front(); q.pop(); // 取出队首元素 in_queue[u] = false; // 标记 u 不在队列中 for (auto v : g[u]) { // 枚举 u 的出边 if (dist[v.first] > dist[u] + v.second) { // 更新 dist[v] dist[v.first] = dist[u] + v.second; if (!in_queue[v.first]) { // 如果 v 不在队列中,则将 v 加入队列 q.push(v.first); in_queue[v.first] = true; } } } } } ``` 解释一下每个部分的作用: 1. 常量 INF 表示距离无限远,MAXN 表示最大点数。 2. g 数组存储有向图,pair 中第一个数表示终点,第二个数表示边权。 3. dist 数组存储源点到每个点的最短距离。 4. in_queue 数组标记每个点是否在队列中。 5. spfa 函数实现 SPFA 算法,s 表示源点编号。 6. 首先将 dist 数组初始化为 INF,表示距离无限远;将 in_queue 数组初始化为 false,表示所有点都不在队列中;将源点 s 加入队列并标记在队列中,并将 dist[s] 初始化为 0。 7. 进行队列操作:取出队首元素 u,枚举 u 的出边,如果可以通过 u 更新 dist[v] 的值,则更新 dist[v] 并将 v 加入队列。当队列为空时,SPFA 算法结束。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值