MOOC浙大数据结构 — 07-图6 旅游规划 (25分)

有了一张自驾旅游路线图，你会知道城市间的高速公路长度、以及该公路要收取的过路费。现在需要你写一个程序，帮助前来咨询的游客找一条出发地和目的地之间的最短路径。如果有若干条路径都是最短的，那么需要输出最便宜的一条路径。

输入格式:

输入说明：输入数据的第1行给出4个正整数$N$、$M$、$S$、$D$，其中$N$（$2\le N\le 500$）是城市的个数，顺便假设城市的编号为0~($N-1$)；$M$是高速公路的条数；$S$是出发地的城市编号；$D$是目的地的城市编号。随后的$M$行中，每行给出一条高速公路的信息，分别是：城市1、城市2、高速公路长度、收费额，中间用空格分开，数字均为整数且不超过500。输入保证解的存在。

输出格式:

在一行里输出路径的长度和收费总额，数字间以空格分隔，输出结尾不能有多余空格。

输入样例:

4 5 0 3
0 1 1 20
1 3 2 30
0 3 4 10
0 2 2 20
2 3 1 20

输出样例:

3 40

#include <cstdio>
#include <vector>
#include <algorithm>

#define MAX 500 + 10
#define INF 0x3fffffff

using namespace std;

typedef struct {
    int l;
    int c;
} Edge;

Edge MGraph[MAX][MAX];

int minCost;
int collected[MAX];
int dist[MAX];
vector<int> tempPath;
vector<int> path[MAX];
vector<int> bestPath;

void initMGraph() {
    for( int i = 0; i < MAX; i++ ) {
        for( int j = 0; j < MAX; j++ ) {
            MGraph[i][j].l = MGraph[i][j].c = INF;
            MGraph[j][i].l = MGraph[j][i].c = INF;
        }
    }
}

void Dijkstra( int n, int start ) {
    fill( dist, dist + MAX, INF );
    dist[start] = 0;
    for( int i = 0; i < n; i++ ) {
        int u = -1, MIN = INF;
        for( int j = 0; j < n; j++ ) {
            if( collected[j] == 0 && dist[j] < MIN ) {
                u = j;
                MIN = dist[j];
            }
        }
        if( u == -1 ) return;
        collected[u] = 1;
        for( int v = 0; v < n; v++ ) {
            if( collected[v] == 0 && MGraph[u][v].l != INF ) {
                if( dist[u] + MGraph[u][v].l < dist[v] ) {
                    dist[v] = dist[u] + MGraph[u][v].l;
                    path[v].clear();
                    path[v].push_back( u );
                } else if( dist[u] + MGraph[u][v].l == dist[v] ) {
                    path[v].push_back( u );
                }
            }
        }
    }
}

void DFS( int v, int s ) {
    if( v == s ){       //递归边界，到达起点
        int sum = 0;
        tempPath.push_back( s );
        for(int i = tempPath.size()-1;i;i--){
            sum += MGraph[tempPath[i]][tempPath[i-1]].c;
        }
        if(sum>0&&sum<minCost){
            bestPath = tempPath;
            minCost = sum;
        }
        tempPath.pop_back();
        return;
    }
    tempPath.push_back(v);
    for(int i = 0;i < path[v].size();i++){
        DFS( path[v][i], s );
    }
    tempPath.pop_back();
}
int main() {
    //freopen( "123.txt", "r", stdin );
    initMGraph();
    int n, m, s, d;
    minCost = 0x3fffffff;
    scanf( "%d%d%d%d", &n, &m, &s, &d );

    int a, b, length, cost;
    for( int i = 0; i < m; i++ ) {
         scanf( "%d%d%d%d", &a, &b, &length, &cost );
         MGraph[a][b].l = length;
         MGraph[a][b].c = cost;
         MGraph[b][a].l = length;
         MGraph[b][a].c = cost;
    }

    Dijkstra( n, s );
    DFS( d, s );
    ////for( int i = 0; i < n; i++ ) printf( "%d ", dist[i] );
    //printf( "\n" );
    //for( int i = 0; i < bestPath.size(); i++ ) printf( "%d ", bestPath[i] );
    //printf( "\n" );
    printf( "%d %d", dist[d], minCost );
}