dijkstra算法代码实现

//最短路径dijkstra算法  图的下标从1开始
//#include"stdafx.h"
#include<iostream>
#define size 1010
#define INF 0x3fffffff
using namespace std;
struct Graph {
int vexNum;//顶点数
int edge;//边数
int map[size][size];//图的邻接矩阵
};
Graph g;
int pre[size];//pre[v]的值为前驱顶点下标
int length[size];//length[v]表示startPos到v的最短路径长度和

void dijkstra(int startPos) {
int i, j;
int min;
bool final[size];//标记是否求得顶点startPos至w的最短路径
//初始化数据
for (i = 1; i <= g.vexNum; i++) {
final[i] = false;
length[i] = g.map[startPos][i];
pre[i] = startPos;
}
length[startPos] = 0;
final[startPos] = true;
//开始主循环，每次求得startPos到某个v顶点的最短路径,并 加入v到集合S
for (i = 1; i <= g.vexNum; i++) {
if (i == startPos)
continue;
min = INF;
//寻找length数组中下标最小的值与下标
for (j = 1; j <= g.vexNum; j++) {
if (!final[j] && length[j] < min) {
min = length[j];
i = j;
}
}
final[i] = true;
//更新length和pre
for (j = 1; j <= g.vexNum; j++) {
//如果经过i顶点的路径比现在这条路径的长度短的话
if (!final[j] && (min + g.map[i][j] < length[j])) {
//说明找到了更短的路径，修改length[j]和pre[j]
length[j] = min + g.map[i][j];
pre[j] = i;
}
}
}
}

void init() {
for (int i = 1; i <= g.vexNum; i++) {
for (int j = 1; j <= g.vexNum; j++) {
if (i == j) {
g.map[i][j] = 0;
}
else {
g.map[i][j] = INF;
}
}
}
}

int main() {
//输入顶点数、边数
while (cin >> g.vexNum >> g.edge && g.vexNum) {
init();
int startPos, endPos, value;
for (int i = 0; i < g.edge; i++) {
cin >> startPos >> endPos >> value;
g.map[startPos][endPos] = g.map[startPos][endPos] = value;
}
//输入最短路径的开始节点与终端节点
cout << "输入最短路径的开始节点与终端节点" << endl;
cin >> startPos >> endPos;
dijkstra(startPos);
cout << length[endPos] << endl;
}
return 0;
}