力扣 787. K 站中转内最便宜的航班 dp

最新推荐文章于 2024-10-01 18:52:56 发布

csu_xiji

最新推荐文章于 2024-10-01 18:52:56 发布

阅读量131

点赞数

分类专栏：力扣 dp 动态规划文章标签：动态规划

本文链接：https://blog.csdn.net/xiji333/article/details/119943784

版权

力扣同时被 2 个专栏收录

282 篇文章 2 订阅

订阅专栏

dp 动态规划

129 篇文章 2 订阅

订阅专栏

https://leetcode-cn.com/problems/cheapest-flights-within-k-stops/
在这里插入图片描述
思路：经典 $d p$ 题想歪wa半天。考虑 $dp_{i,j}$ 表示经过 $i$ 条边后到达点 $j$ 的最便宜的价格。显然有 $dp_{0,src}=0$ ，最终答案就是 $min_{i,dst}$ ，其中 $0 < = i < = k + 1$ 。怎么转移呢？其实方式有很多种，最直观的方式就是对于每个 $i$ ，遍历每一条边计算：
$dp_{i,j}=min(dp_{i,j},dp_{i-1,k}+cost_{k,j})\ if\ E_{k,j}\ in\ E$
也可以仅对于 $i - 1$ 中出现的状态进行扩展。

class Solution {
public:
    int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int k) {
        using pr=pair<int,int>;
        vector<vector<pr>> graph(n);
        for(vector<int>& flight:flights)
            graph[flight[0]].emplace_back(flight[1],flight[2]);
        int inf=0x3f3f3f3f;
        vector<vector<int>> dp(k+2,vector<int>(n,inf));
        dp[0][src]=0;
        int ans=inf;
        for(int i=0;i<=k;i++)
        {
            for(int j=0;j<n;j++)
            {
                if(dp[i][j]==inf)
                    continue;
                for(pr& nxt:graph[j])
                    dp[i+1][nxt.first]=min(dp[i+1][nxt.first],dp[i][j]+nxt.second);
            }
            ans=min(ans,dp[i][dst]);
        }
        ans=min(ans,dp[k+1][dst]);
        return ans==inf?-1:ans;
    }
};