PTA数据结构练习题——旅游规划

题目介绍:

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

输入格式:
输入说明：输入数据的第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>

using namespace std;

vector<int> visited;
int main()
{
    int N, M, S, D;
    cin >> N >> M >> S >> D;
    int c1, c2, c3, c4;
    vector<vector<int>> pay(N, vector<int> (N));
    vector<vector<int>> distance(N, vector<int> (N));
    for(int i = 0; i < M; ++i) {
        cin >> c1 >> c2 >> c3 >> c4;
        distance[c1][c2] = c3;
        distance[c2][c1] = c3;
        pay[c1][c2] = c4;
        pay[c2][c1] = c4;
    }
    //vector<int> path(N);
    vector<int> cost(N);
    vector<int> dis(N);
    for(int i = 0; i < N; ++i) {
        if(S == i) {
            dis[i] = 0;
            cost[i] = 0;
        }
        else if(distance[S][i]) {
            dis[i] = distance[S][i];
            cost[i] = pay[S][i];
        }
            
        else {
            dis[i] = 1e9;
        }
    }
    visited.resize(N);
    visited[S] = 1;
    auto cmp = [&](const int &u, const int &v) {
        return dis[u] > dis[v];
    };
    priority_queue<int, vector<int>, decltype(cmp)> pq(cmp);
    for(int i = 0; i < N; ++i) {
        if(i != S)
            pq.emplace(i);
    }
    while(true) {
        if(pq.empty())
            break;
        int V = pq.top();
        pq.pop();
        visited[V] = 1;
        for(int i = 0; i < N; ++i) {
            if(visited[i] == 0 && distance[V][i]) {
                if(dis[V] + distance[V][i] < dis[i]) {
                    dis[i] = dis[V] + distance[V][i];
                    cost[i] = cost[V] + pay[V][i];
                    //path[i] = V;
                }
                else if(dis[V] + distance[V][i] == dis[i]) {
                    if(cost[V] + pay[V][i] < cost[i]) {
                        cost[i] = cost[V] + pay[V][i];
                        //path[i] = V;
                    }
                }
            }
        }
    }
    cout << dis[D] << " " << cost[D] << endl;
    return 0;
}