Dijkstra算法改动一下,若路长相等且花费更少,更新path和cost
#include <iostream>
#define ERROR -1
using namespace std;
/*----------------------------图的邻接矩阵表示法----------------------------*/
// 直接复制模板, 再加以修改
#define MaxVertexNum 500 /* 最大顶点数设为500 */
#define INFINITY 65535 /* ∞设为双字节无符号整数的最大值65535*/
typedef int Vertex; /* 用顶点下标表示顶点, 为整型 */
typedef int WeightType; /* 边的权值设为整型 */
// typedef char DataType; /* 顶点存储的数据类型设为字符型 */
/* 边的定义 */
typedef struct ENode *PtrToENode;
struct ENode {
Vertex V1, V2; /* 有向边<V1, V2> */
WeightType Weight; /* 权重 */ // 长度
int Cost; // 收费额
};
typedef PtrToENode Edge;
/* 图结点的定义 */
typedef struct GNode *PtrToGNode;
struct GNode {
int Nv; /* 顶点数 */
int Ne; /* 边数 */
WeightType G[MaxVertexNum][MaxVertexNum]; /* 邻接矩阵 */ // 存长度
int Cost[MaxVertexNum][MaxVertexNum]; // 存收费额
// DataType Data[MaxVertexNum]; /* 存顶点的数据 */
// /* 注意:很多情况下,顶点无数据,此时Data[]可以不用出现 */
};
typedef PtrToGNode MGraph; /* 以邻接矩阵存储的图类型 */
/* 初始化一个有VertexNum个顶点但没有边的图 */
MGraph CreateGraph( int VertexNum ) {
Vertex V, W;
MGraph Graph;
Graph = new GNode; /* 建立图 */
Graph->Nv = VertexNum;
Graph->Ne = 0;
/* 初始化邻接矩阵 */
/* 注意:这里默认顶点编号从0开始,到(Graph->Nv - 1) */
for (V=0; V<Graph->Nv; V++)
for (W=0; W<Graph->Nv; W++) {
Graph->G[V][W] = INFINITY;
Graph->Cost[V][W] = 0;
}
return Graph;
}
void InsertEdge( MGraph Graph, Edge E ) {
/* 插入边 <V1, V2> */
Graph->G[E->V1][E->V2] = E->Weight;
Graph->Cost[E->V1][E->V2] = E->Cost;
/* 若是无向图,还要插入边<V2, V1> */
Graph->G[E->V2][E->V1] = E->Weight;
Graph->Cost[E->V2][E->V1] = E->Cost;
}
MGraph BuildGraph( int N, int M ) {
MGraph Graph;
Edge E;
// Vertex V;
// int Nv, i;
// cin >> Nv; /* 读入顶点个数 */
Graph = CreateGraph(N); /* 初始化有Nv个顶点但没有边的图 */
// cin >> Graph->Ne; /* 读入边数 */
Graph->Ne = M;
if ( Graph->Ne != 0 ) { /* 如果有边 */
E = new ENode; /* 建立边结点 */
/* 读入边,格式为"起点 终点 长度 收费",插入邻接矩阵 */
for ( int i=0; i < Graph->Ne; i++ ) {
cin >> E->V1 >> E->V2 >> E->Weight >> E->Cost;
/* 注意:如果权重不是整型,Weight的读入格式要改 */
InsertEdge( Graph, E );
}
}
// /* 如果顶点有数据的话,读入数据 */
// for (V=0; V<Graph->Nv; V++)
// cin >> Graph->Data[V];
return Graph;
}
/*--------------------邻接矩阵存储 - 有权图的单源最短路算法--------------------*/
Vertex FindMinDist( MGraph Graph, int dist[], int collected[] )
{ /* 返回未被收录顶点中dist最小者 */
Vertex MinV, V;
int MinDist = INFINITY;
for (V=0; V<Graph->Nv; V++) {
if ( collected[V]==false && dist[V]<MinDist) {
/* 若V未被收录,且dist[V]更小 */
MinDist = dist[V]; /* 更新最小距离 */
MinV = V; /* 更新对应顶点 */
}
}
if (MinDist < INFINITY) /* 若找到最小dist */
return MinV; /* 返回对应的顶点下标 */
else return ERROR; /* 若这样的顶点不存在,返回错误标记 */
}
void Dijkstra( MGraph Graph, int dist[], int cost[], int path[], Vertex S ) { // S: 源点
int collected[MaxVertexNum];
Vertex V, W;
/* 初始化:此处默认邻接矩阵中不存在的边用INFINITY表示 */
for ( V=0; V<Graph->Nv; V++ ) {
dist[V] = Graph->G[S][V];
if ( dist[V]<INFINITY ) {
path[V] = S;
cost[V] = Graph->Cost[S][V];
}
else {
path[V] = -1;
cost[V] = 0;
}
collected[V] = false;
}
/* 先将起点收入集合 */
dist[S] = 0;
collected[S] = true;
while (1) {
/* V = 未被收录顶点中dist最小者 */
V = FindMinDist( Graph, dist, collected );
if ( V==ERROR ) /* 若这样的V不存在 */
break; /* 算法结束 */
collected[V] = true; /* 收录V */
for( W=0; W<Graph->Nv; W++ ) /* 对图中的每个顶点W */
/* 若W是V的邻接点并且未被收录 */
if ( collected[W]==false && Graph->G[V][W]<INFINITY ) {
// if ( Graph->G[V][W]<0 ) /* 若有负边 */
// return false; /* 不能正确解决,返回错误标记 */
/* 若收录V使得dist[W]变小 */
if ( dist[V]+Graph->G[V][W] < dist[W] ) {
dist[W] = dist[V]+Graph->G[V][W]; /* 更新dist[W] */
path[W] = V; /* 更新S到W的路径 */
cost[W] = cost[V] + Graph->Cost[V][W];
}
else if( dist[V]+Graph->G[V][W] == dist[W] && cost[V]+Graph->Cost[V][W] < cost[W] ) {
path[W] = V;
cost[W] = cost[V] + Graph->Cost[V][W];
}
}
} /* while结束*/
// return true; /* 算法执行完毕,返回正确标记 */
}
int main() {
int N, M, S, D; // N(2≤N≤500):城市的个数, 编号0~(N−1) M:高速公路的条数 S:出发城市编号 D:目的城市编号
cin >> N >> M >> S >> D;
MGraph G = BuildGraph(N, M); // 邻接矩阵
int dist[MaxVertexNum], cost[MaxVertexNum], path[MaxVertexNum];
Dijkstra(G, dist, cost, path, S);
cout << dist[D] << " " << cost[D];
}