输入样例:
第一行:顶点数、边数、起始顶点编号
第2-边数+1行:顶点a,顶点b,边权
输出:输出所有顶点的最短距离
注:为有向带权图
代码:
#include <stdio.h>
#include <algorithm>
using namespace std;
const int maxn = 1000;
const int INF = 100000000;
int n,m,s,G[maxn][maxn];
int d[maxn];
bool vis[maxn] = {false};
void Dijkstra(int s){
fill(d,d+maxn,INF);
d[s] = 0;
for(int i = 0;i <n;i++){
int u = -1,MIN = INF;
for(int j = 0;j <n;j++){
if(vis[j] == false && d[j]<MIN){
u = j;
MIN = d[j];
}
}
if(u == -1) return;
vis[u] = true;
for(int v = 0; v<n;v++){
if(vis[v] == false && G[u][v] != INF && d[u]+G[u][v] <d[v]){
d[v] = d[u]+G[u][v];
}
}
}
}
int main(){
int u,v,w;
scanf("%d %d %d",&n,&m,&s); //顶点个数,边数,起点编号
fill(G[0],G[0]+maxn*maxn,INF); //初始化图G
for(int i = 0;i <m;i++){
scanf("%d%d%d",&u,&v,&w); //输入u,v,以及u->v的边权
G[u][v] = w;
}
Dijkstra(s); //算法入口
for(int i = 0;i <n;i++){
printf("%d ",d[i]); //输出所有顶点的最短距离
}
return 0;
}
/*
6 8 0
0 1 1
0 3 4
0 4 4
1 3 2
2 5 1
3 2 2
3 4 3
4 5 3
*/