leetcode K站中转最便宜的航班

有 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, vector<vector<int>>& flights, int src, int dst, int K) {
        unordered_map<int,vector<pair<int,int>>> m;
        for(auto&flight:flights)
        {
            m[flight[0]].push_back({flight[1],flight[2]});
            //将键为fight[0],值为pair型{flight[1],flight[2]}的元素加入vector末尾,构成图
        }
        int ans=0x7fffffff;//设为整型最大值
        bfs(m,src,dst,K,ans);
        return ans==0xfffffff?-1:ans;
    }
    void bfs(unordered_map<int,vector<pair<int,int>>> &m,int src,int dst,int K,int &ans)
    //ans指飞机进行到src,已经有的路程再加上某一条从src到下一点的路程,在后续反复被替换
    //广度优先法
    {
        queue<pair<int,int>> q;
        q.push({src,0});
        //将原点入栈
        int step=0;
        //step记录relax松弛函数的次数;
        while(!q.empty()&&step<=K)
        {
            int n=q.size();
            //记录栈中元素
            for(int i=0;i<n;i++)
            {
                pair<int,int> p=q.front();
                q.pop();
                for(auto &next:m[p.first])
                //寻找以p.first开头的边
                {
                    if(next.second+p.second>=ans)
                    //边权值+dp[i](p.second)大于dp[i+1](ans)
                        continue;
                    if(next.first==dst)
                    {
                        ans=min(ans,next.second+p.second);
                    }
                    else
                        q.push({next.first,next.second+p.second});
                        //取代ans,并将此元素进栈

                }

            }
            step++;
        }




    }




};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值