题目
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/cheapest-flights-within-k-stops
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
有 n 个城市通过一些航班连接。给你一个数组 flights ,其中 flights[i] = [fromi, toi, pricei] ,表示该航班都从城市 fromi 开始,以价格 pricei 抵达 toi。
现在给定所有的城市和航班,以及出发城市 src 和目的地 dst,你的任务是找到出一条最多经过 k 站中转的路线,使得从 src 到 dst 的 价格最便宜 ,并返回该价格。 如果不存在这样的路线,则输出 -1。
示例
输入:
n = 3, edges = [[0,1,100],[1,2,100],[0,2,500]]
src = 0, dst = 2, k = 1
输出: 200
解释:
城市航班图如下
从城市 0 到城市 2 在 1 站中转以内的最便宜价格是 200,如图中红色所示。
思路(dp)
答案
/**
* @param {number} n
* @param {number[][]} flights
* @param {number} src
* @param {number} dst
* @param {number} k
* @return {number}
*/
var findCheapestPrice = function(n, flights, src, dst, k) {
const INF = 10000 * 101 + 1;
const dp = new Array(n).fill(0).map(() => new Array(k + 2).fill(INF));;//dp[i][t]表示到第i个节点,走了t次
dp[src][0]=0//从src出发到src,走0次,花费0。从其他地方出发走0次到src都是INF
for(let count=1;count<=k+1;++count){
//最多经过k个点会走k+1次,那么count最大为k+1
//现在通过fight判断如果只走count次,dp的结果
for (const flight of flights) {
const j = flight[0], i = flight[1], cost = flight[2];///i--->j 花费cost
dp[i][count] = Math.min(dp[i][count], dp[j][count - 1] + cost);
}
}
let ans = INF;
for (let count = 1; count <= k + 1; ++count) {
ans = Math.min(ans, dp[dst][count]);
}
return (ans == INF ? -1 : ans);
};