题目:K 站中转内最便宜的航班
有 n 个城市通过 m 个航班连接。每个航班都从城市 u 开始,以价格 w 抵达 v。
现在给定所有的城市和航班,以及出发城市 src 和目的地 dst,你的任务是找到从 src 到 dst 最多经过 k 站中转的最便宜的价格。 如果没有这样的路线,则输出 -1。
思路:
本题其实有个隐藏设定,即目的地 dst 是数字最大的城市,或相对大的城市
将距离花费存到一个二维数组中
依次将出发地之后可达的路径加入优先队列 PriorityQueue 中
判断是否超过 K 站中转,是则结束``
是否到达目的地,到达则跳出
class Solution {
public int findCheapestPrice(int n, int[][] flights, int src, int dst, int K) {
int[][] g = new int[n][n];//记录的是目的地
for(int[] f : flights)
{
g[f[0]][f[1]] = f[2];
}
PriorityQueue<int[]> heap = new PriorityQueue<>((a,b)->a[0] - b[0]);
//集合的参数是一个comparator的lambda表达式,默认升序
heap.offer(new int[]{0, src, K + 1});//想集合添加一个记录费用、起点和中转站+1的数组
//K + 1是还可以走过站点的个数
while(!heap.isEmpty())//数组为空直接返回-1
{
int[] cur = heap.poll();//得到集合当中添加的数组
int price = cur[0], place = cur[1], remainStops = cur[2];
if(place == dst)//起点等于v中点
return price;//返回0费用0
if(remainStops > 0)//中转次数》0(至少执行一次,因为remain====k+1)
{
for(int i = 0; i < n; i++)//小于城市数量
{
if(g[place][i] > 0)//表示起点----终点的中转路线是否存在
{
heap.offer(new int[]{price + g[place][i], i, remainStops - 1});
//如果存在 计算路费、起点的值、中转站-1
}
}
}
}
return -1;
}
}