题目

有 n 个城市通过一些航班连接。给你一个数组 flights ，其中 flights[i] = [from(i), to(i), price(i)] ，表示该航班都从城市 fromi 开始，以价格 price(i) 抵达 to(i)。

现在给定所有的城市和航班，以及出发城市 src 和目的地 dst，你的任务是找到出一条最多经过 k 站中转的路线，使得从 src 到 dst 的价格最便宜，并返回该价格。如果不存在这样的路线，则输出 -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

关键思路

本质为求最短路径，这里使用到A*寻路算法。

代码实现

class Solution(object):
    def findCheapestPrice(self, n, flights, src, dst, k):
        """
        :type n: int  number of cities
        :type flights: List[List[int]]
        :type src: int
        :type dst: int
        :type k: int  max sites
        :rtype: int  min price or -1
        """

        # A-star
        cur = src
        open_list = [ [float("inf"), -1 ] for i in range(n)]
        open_list[0][0] = 0

        while cur!=dst:
            for flight in flights:
                if cur == flight[0]:  # match
                    if open_list[cur][0]+flight[2] <= open_list[flight[1]][0]:  # compare
                        if flight[1] <= dst and open_list[flight[1]][1] < k: # compare the number
                            open_list[flight[1]][0] = open_list[cur][0]+flight[2]  # update
                            open_list[flight[1]][1] += 1
            cur = cur+1
        
        return open_list[dst][0] if open_list[dst][0] != float("inf") else -1
    

if __name__ == "__main__":
    n = input()
    edges = input()
    src = input()
    dst = input()
    k = input()
    obj = Solution()
    result = obj.findCheapestPrice(n, edges, src, dst, k)
    print(result)