题目:
有 n 个城市通过 m 个航班连接。每个航班都从城市 u 开始,以价格 w 抵达 v。
现在给定所有的城市和航班,以及出发城市 src 和目的地 dst,你的任务是找到从 src 到 dst 最多经过 k 站中转的最便宜的价格。 如果没有这样的路线,则输出 -1。
思路:
动态规划:
解答:
class Solution:
def findCheapestPrice(self, n: int, flights: List[List[int]], src: int, dst: int, K: int) -> int:
# dp[k][d] 记录经过k次转机后到达d城市的最小花费
dp = [[float('inf')]*n for i in range(K+1)]
# 初始化0次转机直达的花费
for s,d,p in flights:
if s == src :
dp[0][d] = p
for k in range(1,K+1):
for s,d,p in flights:
# 状态转移
dp[k][d] = min(dp[k-1][d],dp[k-1][s]+p,dp[k][d])
return dp[K][dst] if dp[K][dst] != float('inf') else -1