算法步骤:
a.初始时,S只包含源点,即S={v},v的距离为0。U包含除v外的其他顶点,即:U={其余顶点},若v与U中顶点u有边,则<u,v>正常有权值,若u不是v的出边邻接点,则<u,v>权值为∞。
b.从U中选取一个距离v最小的顶点k,把k,加入S中(该选定的距离就是v到k的最短路径长度)。
c.以k为新考虑的中间点,修改U中各顶点的距离;若从源点v到顶点u的距离(经过顶点k)比原来距离(不经过顶点k)短,则修改顶点u的距离值,修改后的距离值的顶点k的距离加上边上的权。
d.重复步骤b和c直到所有顶点都包含在S中。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <cstdlib>
using namespace std;
#define INF 0x3f3f3f3f
#define MAXNUM 110
int dist[MAXNUM];//存最短距离
int N,M;//点数N,边数M
int map[MAXNUM][MAXNUM];
struct node{
int x,len;
node(int a,int b):x(a),len(b){}
};
struct cmp{
bool operator ()(node a,node b){
if(a.len == b.len)return a.x > b.x;
else return a.len > b.len;
}
};
void Dijkstra(int S){
memset(dist,INF,sizeof(dist));
dist[S] = 0;
priority_queue<node,vector<node>,cmp> Q;
Q.push(node(S,0));
while(!Q.empty()){
node now = Q.top();
Q.pop();
for(int i=1 ; i<=N ; i++){
if(dist[i] > dist[now.x] + map[now.x][i]){
dist[i] = dist[now.x]+map[now.x][i];
Q.push(node(i,dist[i]));
}
}
}
}
int main(){
while(cin>>N>>M && N && M){
memset(map,INF,sizeof(map));
while(M--){
int a,b,c;
cin>>a>>b>>c;
map[a][b] = map[b][a] = c;/*有向图map[a][b] = c*/
}
int S;//起点
cin>>S;
Dijkstra(S);
}
return 0;
}