每日力扣-动态规划-k站点问题

93 阅读1分钟

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

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

输入: n = 3, edges = [[0,1,100],[1,2,100],[0,2,500]] src = 0, dst = 2, k = 1 输出: 200

解题思路:i到j经过不多于K站,则可以到j上一站,然后中转一次到j。

class Solution {
public:
    int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {
        const int INF=0x3f3f3f3f;
        vector<int> dp(n,INF);
        dp[src] = 0;
        for(int i = 0 ; i <= K ; i++)
        {
            vector<int> pre = dp;
            for(auto flight : flights)
            {
                dp[flight[1]] = min(dp[flight[1]] , pre[flight[0]] + flight[2]);
            }
        }
        return dp[dst]==INF?-1:dp[dst];
    }
};