PTA 7-10（圖） 旅遊規劃 最短路問題

7-10（圖） 旅遊規劃 （25 分）

有了一張自駕旅遊路線圖，你會知道城市間的高速公路長度、以及該公路要收取的過路費。如今須要你寫一個程序，幫助前來諮詢的遊客找一條出發地和目的地之間的最短路徑。若是有若干條路徑都是最短的，那麼須要輸出最便宜的一條路徑。

輸入格式:

輸入說明：輸入數據的第1行給出4個正整數N、M、S、D，其中N（2≤N≤500）是城市的個數，順便假設城市的編號爲0~(N−1)；M是高速公路的條數；S是出發地的城市編號；D是目的地的城市編號。隨後的M行中，每行給出一條高速公路的信息，分別是：城市一、城市二、高速公路長度、收費額，中間用空格分開，數字均爲整數且不超過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

思路：簡單的最短路問題，dijkstra算法稍加修改就能夠過掉

AC代碼：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstdio>
#include <malloc.h>

#define INF 0x3f3f3f3f
#define FRER() freopen("in.txt", "r", stdin)
#define FREW() freopen("out.txt", "w", stdout)

using namespace std;

const int maxn = 500 + 5;

struct edge{
    int v, l, w;
    edge(int v, int l, int w):v(v), l(l), w(w) {}
    friend bool operator < (edge a, edge b) {
        return a.l > b.l;
    }
};

vector<edge> g[maxn];
int n, m, s, d, u, v, l, w, vis[maxn], len[maxn], price[maxn];

void dijkstra() {
    memset(len, INF, sizeof(len));
    priority_queue<edge> q;
    q.push(edge(s, 0, 0));
    while(!q.empty()) {
        edge tmp = q.top(); q.pop();
        if(vis[tmp.v]) {
            if(tmp.l > len[tmp.v] || (tmp.l == len[tmp.v] && tmp.w >= price[tmp.v]))
                continue;
        }
        len[tmp.v] = tmp.l;
        price[tmp.v] = tmp.w;
        vis[tmp.v] = true;
        for(int i = 0; i < g[tmp.v].size(); ++i) {
            if(tmp.l + g[tmp.v][i].l < len[g[tmp.v][i].v] || (tmp.l + g[tmp.v][i].l == len[g[tmp.v][i].v] && tmp.w + g[tmp.v][i].w < price[g[tmp.v][i].v]))
            q.push(edge(g[tmp.v][i].v, tmp.l + g[tmp.v][i].l, tmp.w + g[tmp.v][i].w));
        }
    }
}

int main()
{
    cin >> n >> m >> s >> d;
    while(m--) {
        cin >> u >> v >> l >> w;
        g[u].push_back(edge(v, l, w));
        g[v].push_back(edge(u, l, w));
    }
    dijkstra();
    cout << len[d] << ' ' << price[d] << endl;
    return 0;
}