一、思路
该算法和prim算法(最小生成树)类似,区别在于更新当前所能选择的顶点的数组(此处用lowcost代替)时,prim是直接用新加入顶点所能到达的未被访问过的顶点的权重更新lowcost数组,而dijkstra是用新加入顶点所能到达的未被访问过的顶点的权重加上当前新加入顶点所走过的权重更新lowcost数组。用代码来表示就是:
/* prim */
for(int i = 0; i < vertnum; ++i){ // 遍历lowcost数组
if(visit[i] == -1 && Edge[u][i] != INT_MAX){ // 若新加入顶点存在所能到达的未被访问的顶点
if(Edge[u][i] < lowcost[i]){ // 若加入该新顶点后能到达的顶点所需要的权重小于未加入该顶点前所能到达的顶点的权重
lowcost[i] = Edge[u][i]; // 更新lowcost数组
}
}
}
/* dijkstra */
for(int i = 0; i < vertnum; ++i){ // 遍历lowcost数组
if(visit[i] == -1 && Edge[u][i] != INT_MAX){ // 若新加入顶点存在所能到达的未被访问的顶点
if(lowcost[u] + Edge[u][i] < lowcost[i]){ // 若加入该新顶点后能到达的顶点所需要的权重加上到达该新顶点的权重小于未加入该顶点前所能到达的未被访问的顶点的权重
lowcost[i] = Edge[u][i] + lowcost[u]; // 更新lowcost数组
}
}
}
二、代码
void Dijkstra(int v0, UDG& G){
int dist[G.vernum]; // 从v0到每个点的的距离数组(也就是lowcost数组)
int visit[G.vernum]; // 访问数组
for(int i = 0; i < G.vernum; ++i){ // 初始化访问数组和dist数组
dist[i] = INT_MAX;
visit[i] = -1;
}
dist[v0] = 0; // 将v0计入dist数组
visit[v0] = 0; // 将v0设置为已访问
ArcNode *p = G.vers[v0].firarc; // 指针p指向v0的邻接顶点,p->adjver是邻接顶点的下标,p->weight是到达邻接顶点所需要的权重
while(p){ // 使用v0更新dist数组
dist[p->adjver] = p->weight;
p = p->nextarc;
}
for(int i = 0; i < G.vernum - 1; ++i){ // 总共循环n - 1次,因为已经更新了v0顶点
int min = INT_MAX; // 初始最小值默认为最大
int u = -1; // 所能到达的权重最小的顶点的下标
for(int j = 0; j < G.vernum; ++j){ // 搜索dist数组,找出当前最小权重所能到达的顶点并用u记录该顶点下标
if(visit[j] == -1 && dist[j] < min){
u = j;
min = dist[j];
}
}
if(u != -1){ // 若已找到当前最短路径所能到达的顶点
visit[u] = 0; // 记录下标为u的顶点已被访问
p = G.vers[u].firarc;
while(p){
if(visit[p->adjver] == -1 && dist[u] + p->weight < dist[p->adjver]){
dist[p->adjver] = dist[u] + p->weight;
}
p = p->nextarc;
}
}
}
cout << "result:" << endl;
for(int i = 0; i < G.vernum; ++i){
cout << dist[i] << endl;
}
}
若有错误欢迎大佬批评并指出!