LeetCode #787

Cheapest Flights Within K Stops

1개의 풀이 · C++

문제 원문 보기 ↗

SOLUTION INFO

C++ · main.cpp

main.cpp
class Solution {
public:
    int findCheapestPrice(int N, vector<vector<int>>& flights, int src, int dst, int K) {
        vector<vector<pair<int, int>>> G(N);
        for(int i = 0; i < flights.size(); ++i) {
            int u = flights[i][0], v = flights[i][1], w = flights[i][2];
            G[u].emplace_back(v, w);
        }
        const int INF = 0x3f3f3f3f;
        vector<vector<int>> DP(K + 2, vector<int>(N, INF));
        vector<priority_queue<pair<int, int>>> pqs(K + 1);
        pqs[0].emplace(0, src); DP[0][src] = 0;
        int mn = INF;
        for(int k = 0; k <= K; ++k) {
            while(!pqs[k].empty()) {
                auto [d, cur] = pqs[k].top(); pqs[k].pop();
                if(cur == dst) mn = min(mn, DP[k][cur]);
                if(DP[k][cur] != -d) continue;
                for(const auto &[nxt, w]: G[cur]) {
                    if(DP[k + 1][nxt] > -d + w) {
                        DP[k + 1][nxt] = -d + w;
                        if(k+1<=K) pqs[k+1].emplace(d - w, nxt);
                    }
                }
            }
        }
        mn = min(mn, DP[K + 1][dst]);
        if(mn == INF) mn = -1;
        return mn;
    }
};

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.