PTA 7-9 旅游规划 (25 point(s))

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

输入格式:

输入说明：输入数据的第1行给出4个正整数N、M、S、D，其中N（2≤N≤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 <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define MAXN 501
using namespace std;

int N,M,S,D;
int G[MAXN][MAXN][2] = {0};			//存放城市的[0]长度 [1]路费
int visited[MAXN] = {false};		//标记 
int dist[MAXN];						//距离 
int cost[MAXN];						//花费 

void Dijkstra() {
    // 源点距离和路费为0
    dist[S] = cost[S] = 0;
    
    while(true){
        // 最短距离min 记录距离最短城市编号k
        int MIN = INF, k = -1;
        
        // 找到最短距离城市
        for(int i = 0; i < N; ++i){
            // 当距离更短 && 结点没有访问过
            if(visited[i] == false && dist[i] < MIN){
                k = i;
                MIN = dist[i];
            }
        }
        
        // 如果没有找到最短距离城市，意味当前没有连通的城市
        if(k == -1)
            return;
        
        // 标记城市已访问
        visited[k] = true;
        
        // 找到新城市后 优化距离和路费
        for(int i = 0; i < N; ++i){
            
            // 当城市未访问过
            if(visited[i] == false){
                // 存在更短距离时
                if(dist[k] + G[k][i][0] < dist[i]){
                    dist[i] = dist[k] + G[k][i][0];
                    cost[i] = cost[k] + G[k][i][1];
                }
            
                // 如果距离相等  && 存在更少路费 
                else if(dist[k] + G[k][i][0] == dist[i]){
                    if(cost[k] + G[k][i][1] < cost[i]){
                        dist[i] = dist[k] + G[k][i][0];
                        cost[i] = cost[k] + G[k][i][1];
                    }
                }
            }
        }
    }
}

int main() {
    cin >> N >> M >> S >> D;
    
    // 初始化图
    for(int i = 0; i < N; ++i){
        // 初始化距离以及路费
        cost[i] = INF;
        dist[i] = INF;
        
        for(int j = 0; j < N; ++j){
            // 除对角元
            if(i != j)
                G[i][j][0] = G[i][j][1] = INF;
        }
    }
    
    // 读取城市间的距离以及路费 (无向图)
    int x, y, d, c;
    while(cin >> x >> y >> d >> c){
        G[x][y][0] = G[y][x][0] = d;
        G[x][y][1] = G[y][x][1] = c;
    }
    
    // Dijkstra算法求出最短距离及路费
    Dijkstra();
    
    cout << dist[D] << " " << cost[D];
    return 0;
}

参考代码