LeetCode 787. K 站中转内最便宜的航班(Dijkstra最短路径 + 优先队列)
生活随笔
收集整理的這篇文章主要介紹了
LeetCode 787. K 站中转内最便宜的航班(Dijkstra最短路径 + 优先队列)
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
文章目錄
- 1. 題目
- 2. 解題
1. 題目
有 n 個城市通過 m 個航班連接。每個航班都從城市 u 開始,以價格 w 抵達 v。
現在給定所有的城市和航班,以及出發城市 src 和目的地 dst,你的任務是找到從 src 到 dst 最多經過 k 站中轉的最便宜的價格。
如果沒有這樣的路線,則輸出 -1。
示例 1:
輸入: n = 3, edges = [[0,1,100],[1,2,100],[0,2,500]] src = 0, dst = 2, k = 1 輸出: 200 從城市 0 到城市 2 在 1 站中轉以內的最便宜價格是 200,如圖中紅色所示。示例 2:
輸入: n = 3, edges = [[0,1,100],[1,2,100],[0,2,500]] src = 0, dst = 2, k = 0 輸出: 500 解釋: 城市航班圖如下 從城市 0 到城市 2 在 0 站中轉以內的最便宜價格是 500,如圖中藍色所示。 提示: n 范圍是 [1, 100],城市標簽從 0 到 n - 1. 航班數量范圍是 [0, n * (n - 1) / 2]. 每個航班的格式 (src, dst, price). 每個航班的價格范圍是 [1, 10000]. k 范圍是 [0, n - 1]. 航班沒有重復,且不存在環路來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/cheapest-flights-within-k-stops
著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。
2. 解題
參考:圖Graph–最短路徑算法(Shortest Path Algorithm)
typedef pair<int,pair<int,int>> piii; struct cmp{bool operator()(piii& a, piii& b) const{return a.first > b.first;//價格低的優先} }; class Solution { public:int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {vector<unordered_map<int,int>> g(n);for(auto& f : flights){g[f[0]][f[1]] = f[2];}K++;//可以走K+1步vector<vector<int>> price(n, vector<int>(K+1,INT_MAX));price[src][0] = 0;//在某點,走了幾步的最少花費priority_queue<piii,vector<piii>,cmp> q;q.push({0, {src, 0}});//路徑機票價格總和, 當前點,走了幾步int id, money, ntid, cost, step;while(!q.empty()){money = q.top().first;//之前總的花費id = q.top().second.first;//當前點step = q.top().second.second;//到這走了幾步if(id == dst)//找到目標return price[id][step];q.pop();for(auto it = g[id].begin(); it != g[id].end(); ++it){ntid = it->first;//下一個點cost = it->second;//需要的花費if(step+1 <= K && money+cost < price[ntid][step+1]){ //步數還可以走 && 花費較小price[ntid][step+1] = money+cost;q.push({money+cost, {ntid, step+1}});}}}return -1;} };40 ms 12.2 MB
我的CSDN博客地址 https://michael.blog.csdn.net/
長按或掃碼關注我的公眾號(Michael阿明),一起加油、一起學習進步!
總結
以上是生活随笔為你收集整理的LeetCode 787. K 站中转内最便宜的航班(Dijkstra最短路径 + 优先队列)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: LeetCode 756. 金字塔转换矩
- 下一篇: LeetCode 1135. 最低成本联