787. K 站中转内最便宜的航班

2021/03/24 leetcode 共 880 字,约 3 分钟

title

problem link

有 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]
  • 航班没有重复,且不存在自环

solution

class Solution:
    def findCheapestPrice(self, n: int, flights: List[List[int]], src: int, dst: int, K: int) -> int:
        graph = collections.defaultdict(list)
        for u, v, w in flights:
            graph[u].append((v, w))
        visit = {}
        hq = [(0, src, 0)]
        while hq:
            cost, u, steps = heapq.heappop(hq)
            if (u, steps) in visit:
                continue
            visit[(u, steps)] = 1
            if steps > K + 1:
                continue
            if u == dst:
                return cost
            for v, w in graph[u]:
                if (v, steps+1) not in visit:
                    heapq.heappush(hq, (cost + w, v, steps + 1))
        return -1

Search

    Table of Contents