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

快来打我* 2022-11-05 08:49 143阅读 0赞

有 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
    著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution {
        
        public int findCheapestPrice(int n, int[][] flights, int src, int dst, int K) {
        
            int graph[][]=new int[n][n];
            //利用二维数组做表格样式保存图的路径
            for(int []flight:flights) {
        
            	graph[flight[0]][flight[1]]=flight[2];
            }
            //保存到达各个点的最优路径长度
            Map<Integer, Integer> map=new HashMap<Integer, Integer>();
            PriorityQueue<int[]> queue=new PriorityQueue<int[]>((a,b)->a[0]-b[0]);
            queue.offer(new int[] {
        0,0,src});//第一个元素表示距离，第二个元素表示周转次数（避免死循环），第三个元素为起始点
            
            while(queue.isEmpty()==false) {
        
            	int []head=queue.poll();
            	int cost=head[0];//花费
            	int cnt=head[1];//中转次数
            	int srcplace=head[2];//起始点
            	
            	if(cnt>K+1||cost>map.getOrDefault(cnt*1000+srcplace, Integer.MAX_VALUE))
            		continue;
            	if(srcplace==dst) return cost;
            	
            	//遍历寻找从当前起始点可以到达的终点站，更新相应站点信息。
            	for(int dstplace=0;dstplace<n;dstplace++) {
        
            		if(graph[srcplace][dstplace]>0) {
        
            			int newcost=cost+graph[srcplace][dstplace];
            			if(newcost<map.getOrDefault((cnt+1)*1000+dstplace, Integer.MAX_VALUE)) {
        
            				queue.offer(new int[] {
        newcost,cnt+1,dstplace});
            				map.put((cnt+1)*1000+dstplace,newcost);
            				}
            			}
            		}
            	
            }
            return -1;
    	}
    }