【模板】最短路算法

1.Floyd算法

#include<iostream>
#include<cstdio>
using namespace std;
const int N=1e4+10;
const int inf=0x3f3f3f3f;
int n,m,f[N][N],q;
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			if(i!=j) f[i][j]=inf;
	for(int i=1;i<=m;i++){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		if(u==v){
			f[u][v]=0,f[v][u]=0;
			continue;
		}
		f[u][v]=min(w,f[u][v]),f[v][u]=min(w,f[v][u]);
	}
	for(int k=1;k<=n;k++)
		for(int x=1;x<=n;x++)
			for(int y=1;y<=n;y++)
				f[x][y]=min(f[x][y],f[x][k]+f[k][y]);
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++) printf("%d ",f[i][j]);
		printf("\n");
	}
	return 0;
}

2.Bellman-Ford算法

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define int long long
const int N=5e5+10;
const int inf=0x3f3f3f3f3f3f3f3f;
struct edge{int v,w;};
vector<edge> g[N];
int dis[N];
void write(int x){if(x<0){putchar('-');x=-x;}if(x>9){write(x/10);putchar(x%10+'0');}else{putchar(x+'0');}return;}
int read(){register int x=0,f=1;register char ch;ch=getchar();while(ch>'9'||ch<'0'){if(ch=='-'){f=-f;}ch=getchar();}while(ch<='9'&&ch>='0'){x=x*10+ch-48;ch=getchar();}return x*f;}
bool bellmanford(int n,int s){
	bool flg=false;
	memset(dis,0x3f,sizeof(dis));
	dis[s]=0;
	for(int i=1;i<=n;i++){
		flg=false;
		for(int u=1;u<=n;u++){
			if(dis[u]==inf) continue;
			int len=g[u].size();
			for(int j=0;j<len;j++){
				int tv=g[u][j].v,tw=g[u][j].w;
				if(dis[tv]>dis[u]+tw){
					dis[tv]=dis[u]+tw;
					flg=true;
				}
			}
		}
		if(!flg) break;
	}
	return flg;
}
signed main(){
	int n,m,s;
	n=read(),m=read(),s=read();
	for(int i=1;i<=m;i++){
		int x,y,z;
		x=read(),y=read(),z=read();
		if(x!=y) g[x].push_back({y,z});
		//g[y].push_back({x,z});
	}
	bellmanford(n,s);
	for(int i=1;i<=n;i++) printf("%lld ",dis[i]!=inf?dis[i]:-1);
	return 0;
}

3.SPFA算法

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
using namespace std;
const int N=1e5+10;
const int inf=0x3f3f3f3f;
struct edge{int v,w;};
vector<edge> g[N];
queue<int> q;
int dis[N],cnt[N];
bool vis[N];
bool spfa(int n,int s){
	for(int i=1;i<=n;i++) dis[i]=inf;
	dis[s]=0,vis[s]=true;q.push(s);
	while(!q.empty()){
		int u=q.front();
		q.pop(),vis[u]=false;
		int len=g[u].size();
		for(int i=0;i<len;i++){
			int tv=g[u][i].v,tw=g[u][i].w;
			if(dis[tv]>dis[u]+tw){
				dis[tv]=dis[u]+tw;
				cnt[tv]=cnt[u]+1;
				if(cnt[tv]>=n) return false;
				if(!vis[tv]){
					q.push(tv);
					vis[tv]=true;
				}
			}
		}
	}
	return true;
}
int main(){
	int n,m,s;
	scanf("%d%d%d",&n,&m,&s);
	while(m--){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		g[u].push_back({v,w});
		//g[v].push_back({u,w});
	}
	spfa(n,s);
	for(int i=1;i<=n;i++) printf("%d ",dis[i]==inf?-1:dis[i]);
	printf("\n");
	return 0;
}

4.Dijsktra算法（堆优化）

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
const int N=2e5+10;
const int inf=0x3f3f3f3f;
struct edge{int v,w;};
struct node{int dis,u;bool operator >(const node& x)const{return dis>x.dis;}};
vector<edge> g[N];
priority_queue<node,vector<node>,greater<node> >q;
int dis[N],vis[N];
void dijkstra(int n,int s){
	for(int i=1;i<=n;i++)dis[i]=inf;
	dis[s]=0;q.push({0,s});
	while(!q.empty()){
		int u=q.top().u;
		q.pop();
		if(vis[u]) continue;
		vis[u]=true;
		int len=g[u].size();
		for(int i=0;i<len;i++){
			int tv=g[u][i].v,tw=g[u][i].w;
			if(dis[tv]>dis[u]+tw){
				dis[tv]=dis[u]+tw;
				q.push({dis[tv],tv});
			}
		}
	}
}
int main(){
	int n,m,s;
	scanf("%d%d%d",&n,&m,&s);
	while(m--){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		g[u].push_back({v,w});
	}
	dijkstra(n,s);
	for(int i=1;i<=n;i++) printf("%d ",dis[i]==inf?-1:dis[i]);
	return 0;
}