Dijkstra 模板

算法步骤:

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;
}
原文地址:https://www.cnblogs.com/vocaloid01/p/9514200.html