SPFA+链式前向星 模板

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<queue>
#define inf 999999999
#define MAX 10000
#define begin Begin
using namespace std;
int begin[MAX],next[MAX],to[MAX],w[MAX],edge;
int vis[MAX];
int dist[MAX];
queue <int> Q;
//spfa+链式前向星
void add(int x,int y,int z){
	to[++edge]=y;
	next[edge]=begin[x];
	begin[x]=edge;
	w[edge]=z;
}
void spfa(int x){
	Q.push(x);//入队 
	dist[x]=0;
	vis[x]=1;
	while(!Q.empty()){
		int now=Q.front();
		Q.pop();//出队 
		vis[now]=0;
		for(int i=begin[now];i;i=next[i]){
			int u=to[i];
			int v=w[i];
			if(dist[u]>dist[now]+v){
				dist[u]=dist[now]+v;
				if(!vis[u]){
					Q.push(u);
					vis[u]=1;
				}
			}
		}
	}
}
int main(){
	int i,j,k,m,n;
	int x,y,z;
	scanf("%d%d",&n,&m);//n个点,m条边 
	for(i=1;i<=n;i++)
		dist[i]=inf;
	for(i=1;i<=m;i++){
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
	}
	scanf("%d",&x);//输入访问起点 
	spfa(x);
	for(i=1;i<=n;i++)
		printf("%d ",dist[i]);
	return 0;
}
//测试数据如下 
/*
5 7
1 2 2
1 5 10
2 3 3
2 5 7
3 4 4
4 5 5
5 3 6
1
0 2 5 9 9
*/
/*
5 6
1 2 5
1 3 9
1 4 1
3 5 2
4 3 3
4 5 7
1
0 5 4 1 6
*/
/*
5 5
1 2 20
2 3 30
3 4 20
4 5 20
1 5 100
1
0 20 50 70 90
*/

猜你喜欢

转载自blog.csdn.net/MallowFlower/article/details/81387089