07-图6 旅游规划

旅游规划

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];
}