普通版
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e2;
const int INF = 1<<30;
int g[maxn][maxn];
int n;
int dist[maxn];
void init(){
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++)g[i][j]=INF;
}
}
void ShortestDist(int v=1){
bool reach[maxn]={0};
reach[v]=true;
for(int i=1;i<=n;i++)dist[i]=g[v][i];
dist[v]=0;
for(int i=0;i<n;i++){
int min = INF,index=-1;
for(int j=1;j<=n;j++){
if(!reach[j]&&min>dist[j])min=dist[j],index=j;
}
if(index==-1)return ;
reach[index]=true;
for(int i=1;i<=n;i++){
if(g[index][i]+dist[index]<dist[i])dist[i]=g[index][i]+dist[index];
}
}
}
int main(){
int m;
int a,b,c;
scanf("%d%d",&n,&m);
init();
for(int i=0;i<m;i++){
scanf("%d%d%d",&a,&b,&c);
g[a][b]=g[b][a]=c;
}
ShortestDist(1);
for(int i=1;i<=n;i++){
printf("%d ",dist[i]);
}
}
测试输入
4 4
1 2 2
1 3 10
2 3 5
2 4 1
测试输出
0 2 7 3
使用优先队列
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e2;
const int INF = 1<<30;
typedef pair<int,int> pa;
int g[maxn][maxn];
int n;
int dist[maxn];
void init(){
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++)g[i][j]=INF;
}
}
void ShortestDist(int v=1){
priority_queue<pa,vector<pa>,greater<pa> > p;
bool reach[maxn]={false};
for(int i=1;i<=n;i++){
dist[i]=g[v][i];
p.push(pa(dist[i],i));
}
int rv=1;
reach[v]=true;
dist[v]=0;
while(!p.empty()){
if(rv==n)return;
v = p.top().second;
p.pop();
if(reach[v])continue;
reach[v]=true;
rv++;
for(int i=1;i<=n;i++){
if(!reach[i]&&dist[v]+g[v][i]<dist[i])dist[i]=dist[v]+g[v][i],p.push(pa(dist[i],i));
}
}
}
int main(){
int m;
int a,b,c;
scanf("%d%d",&n,&m);
init();
for(int i=0;i<m;i++){
scanf("%d%d%d",&a,&b,&c);
g[a][b]=g[b][a]=c;
}
ShortestDist(1);
for(int i=1;i<=n;i++){
printf("%d ",dist[i]);
}
}
SPFA
#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e2;
const int INF = 1<<30;
int g[maxn][maxn];
int n;
int dist[maxn];
void init(){
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++)g[i][j]=INF;
}
}
bool SPFA(int v){
for(int i=1;i<=n;i++)dist[i]=INF;
queue<int>q;
int cnt[maxn]={0};
bool flag[maxn]={false};
q.push(v);
flag[v]=true;
cnt[v]++;
dist[v]=0;
while(!q.empty()){
int j = q.front();
q.pop();
flag[j]=false;
for(int i=1;i<=n;i++){
if(g[j][i]+dist[j]<dist[i]){
dist[i]=g[j][i]+dist[j];
if(!flag[i]){
flag[i]=true;
cnt[i]++;
if(cnt[i]>n)return false;
q.push(i);
}
}
}
}
return true;
}
int main(){
int m;
int a,b,c;
scanf("%d%d",&n,&m);
init();
for(int i=0;i<m;i++){
scanf("%d%d%d",&a,&b,&c);
g[a][b]=g[b][a]=c;
}
if(SPFA(1)){
for(int i=1;i<=n;i++){
printf("%d ",dist[i]);
}
}else{
printf("该图存在负环");
}
}
测试输入
4 4
1 2 2
1 3 10
2 3 5
2 4 1
4 4
1 2 2
1 3 10
2 3 -5
2 4 1
测试输出
0 2 7 3
该图存在负环
参考博客
SPFA算法
Floyd
#include<bits/stdc++.h>
using namespace std;
const int maxn =1e2+3;
int g[maxn][maxn];
int n,m;
const int INF = 1<<30;
void init(){
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)g[i][j]=INF;
}
void floyd(){
int s = 1;
for(int k=1;k<=n;k++)
for(int i=1;i<=n;i++)
if(g[i][k]!=INF)
for(int j=1;j<=n;j++)if(g[i][j]>g[i][k]+g[k][j])
g[i][j]=g[i][k]+g[k][j];
}
int main(){
scanf("%d%d",&n,&m);
init();
int a,b,w;
for(int i=0;i<m;i++){
scanf("%d%d%d",&a,&b,&w);
g[a][b]=g[b][a]=w;
}
floyd();
}
Bellman-Ford
#include<bits/stdc++.h>
using namespace std;
const int maxn =1e5+3;
const int INF = 1<<30;
struct edge{int u,v,w;}e[10005];
int n,m;
int pre[maxn];
int dist[maxn];
void path(int s,int t){
if(s==t){printf("%d ",s);return ;}
path(s,pre[t]);
printf(" %d",t);
}
void bellman(int s=1){
for (int i=1;i<=n;++i)dist[i]=INF;
dist[s]=0;
for(int i=1;i<=n;++i){
for(int j=0;j<m;++j){
int x=e[i].u,y=e[i].v;
if(dist[x]>dist[y]+e[j].w){
dist[x]=dist[y]+e[j].w;
pre[x]=y;
}
}
}
}
int main(){
scanf("%d%d",&n,&m);
int a,b,w;
for(int i=0;i<m;i++){
scanf("%d%d%d",&a,&b,&m);
int x = i<<1;
e[x].u=a,e[x].v=b;e[x++].w=w;
e[x].u=b,e[x].v=a;e[x].w=w;
}
m<<=1;
bellman();
}