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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/cheapest-flights-within-k-stops
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    unordered_map<int,vector<pair<int,int>>> graph;
    void dfs(int src,int dst,int K,int cur,int &ans){
       if(src == dst){
           ans = cur;
           return;
       }
        if(K < 0) return;
        for(auto it : graph[src]){
            if(cur+it.second>=ans) continue;
            dfs(it.first,dst,K-1,cur+it.second,ans);
        }
    }
    int findCheapestPrice(int n, vector<vector<int>>& flights, int src, int dst, int K) {
        int ans = INT_MAX;
        for(auto it : flights) graph[it[0]].push_back(make_pair(it[1],it[2]));
        dfs(src,dst,K,0,ans);
        return (ans == INT_MAX)?-1:ans;
    }
};