Dijkstra算法:解决“边权重非负”的加权“有向图”的“单起点”最短路径问题。
思路:以起始点为中心向外层层扩展,广度优先搜索BFS(用queue)直到扩展到终点为止。
787. Cheapest Flights Within K Stops
Input: n = 3, edges = [[0,1,100],[1,2,100],[0,2,500]]
src = 0, dst = 2, k = 1,K: 最多中转次数,求最小的机票总价 Output: 200
凡是图的题,首先很有可能输入是边矩阵,不管是啥吧,都先转化成:1)邻接矩阵,或者2)邻接表。
这个选择一般看点是否已知。若已知且不是sparse的,可以选1邻接矩阵,否则若需要在构建的过程中才能得知有多少个点,则选2邻接表。本题用2邻接表。
图构建好之后,我们从起始点开始,一层一层往外扩展,记录下扩展到了第几层了,由K开始递减,一旦扩展到K层(remaining stops==0),就停止。有点像那个路由的TTL(Time-to-Live)哈哈。
用一个PriorityQueue来存放三元组:
- price: 当前构成的路径总价格(用这个排序)
- city: 下一步要去的城市
- stops:还可以走几步
每次从PriorityQueue中pop出价格最低的那个三元组,作为当前访问的节点。
在这个节点,我们先检查结束条件:
- good: 如果city(下一步要去的城市)就是终点,则停止。
- bad: 只有stop>0我们才往PriorityQueue里添加三元组,当PriorityQueue为空的时候,认为图已经扫完了,不管是否到达good terminal,都停止。
如果不满足结束条件,则遍历当前节点的邻居,把邻居的信息构成三元组,然后添加到PriorityQueue里,pop的时候是小的先pop,这样就能保证权值小的那个探路的尝试更早找到“destination”。
class Solution {
public int findCheapestPrice(int n, int[][] flights, int src, int dst, int K) {
Map<Integer, Map<Integer, Integer>> prices = new HashMap<>();//<from,<to,price>>
//边数组 ==> 带权值的邻接表
for (int[] f : flights) {
if (!prices.containsKey(f[0])) {
prices.put(f[0], new HashMap<>());