PTA 7-10(圖) 旅遊規劃 最短路問題
阿新 • • 發佈:2018-12-08
7-10(圖) 旅遊規劃 (25 分)
有了一張自駕旅遊路線圖,你會知道城市間的高速公路長度、以及該公路要收取的過路費。現在需要你寫一個程式,幫助前來諮詢的遊客找一條出發地和目的地之間的最短路徑。如果有若干條路徑都是最短的,那麼需要輸出最便宜的一條路徑。
輸入格式:
輸入說明:輸入資料的第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
思路:簡單的最短路問題,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) {} friendbool 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; }