leetcode 787. 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].
  • 航班没有重复,且不存在环路

代码

class Solution {
public:
    int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {
        int ans = INT_MAX;
        vector<bool> isVisited(n+1,false);
        unordered_map<int,vector<vector<int>>> graph;
        for(auto flight:flights)
            graph[flight[0]].push_back({flight[1],flight[2]});
        helper(graph,isVisited,K,src,dst,ans,0);
        return ans == INT_MAX?-1:ans;
    }
    void helper(unordered_map<int,vector<vector<int>>> &graph,vector<bool>&isVisited,int K,int cur,int& dst,int &ans,int sum)
    {
        if(cur == dst)
        {
            ans = sum;return;
        }
        if(K < 0)return;
        for(auto node:graph[cur])
        {
            if(isVisited[node[0]] || sum+node[1] > ans)continue;
            isVisited[node[0]] = true;
            helper(graph,isVisited,K-1,node[0],dst,ans,sum+node[1]);
            isVisited[node[0]] = false;
        }
    }
};

结果

执行用时:1976 ms, 在所有 C++ 提交中击败了5.05%的用户
内存消耗:158.1 MB, 在所有 C++ 提交中击败了5.06%的用户

代码

public:
    int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {
        int ans = INT_MAX;
        vector<vector<int>> dp(K+2,vector<int>(n+1,1e9));
        dp[0][src] = 0;
        for(int i = 1; i < K+2; ++i)
        {
            dp[i][src] = 0;
            for(auto flight:flights)
            {
                dp[i][flight[1]] = min(dp[i][flight[1]],dp[i-1][flight[0]]+flight[2]);
            }
        }
        return dp[K+1][dst] == 1e9?-1:dp[K+1][dst];
    }
};

结果

执行用时:208 ms, 在所有 C++ 提交中击败了14.26%的用户
内存消耗:26.1 MB, 在所有 C++ 提交中击败了5.82%的用户
posted @ 2020-11-27 17:06  曲径通霄  阅读(110)  评论(0编辑  收藏  举报