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

题目

K 站中转内最便宜的航班

有 n 个城市通过 m 个航班连接。每个航班都从城市 u 开始，以价格 w 抵达 v。

现在给定所有的城市和航班，以及出发城市 src 和目的地 dst，你的任务是找到从 src 到 dst 最多经过 k 站中转的最便宜的价格。 如果没有这样的路线，则输出 -1。

思路

• 使用广度优先搜索，但是和普通的BFS有点不同，带有一点层次遍历。因为题目中要求最多是k站。队列里面存储的是（pos, cost）。pos代表的上一个节点的位置，cost表示的是当前花费。
• 每次取队头元素，如果到了dst，那么比较结果。否则，和当前的最小值比较，如果当前cost的数值比最小值小，才能入队列。重复层次遍历k次后，退出循环。

代码

class Solution {
public:
    int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {
        queue<pair<int, int>> q;
        q.push({src,0});
        int ans = 0x3f3f3f3f, tot = 0;
        vector<vector<int>> path(n, vector<int>(n,0));
        for(int i = 0;i < flights.size();i++)//解析flights里面(src, dst, cost)到二维矩阵
            path[flights[i][0]][flights[i][1]] = flights[i][2];
        while(tot <= K && !q.empty()){//如果队列当前空或者说层次遍历超过K次
            int len = q.size();
            for(int t = 0;t < len;t++){//层次遍历
                pair<int, int> cur = q.front();
                q.pop();
                int pos = cur.first, cost = cur.second; 
                for(int i = 0; i < path[pos].size();i++ ){
                    if(path[pos][i] > 0 && cost + path[pos][i] < ans) {
                        if(i == dst) ans = min(ans, cost + path[pos][i]);
                        else q.push({i, cost + path[pos][i]});
                    }
                }
            }
            tot++;
        }
        if(ans == 0x3f3f3f3f) return -1;
        else return ans;
    }
};