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

풀이 설명

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

GITHUB COMMUNITY

커뮤니티 평가

GitHub 계정으로 남긴 최신 평가 1건만 반영하며, 데이터는 매일 저장소에 동기화됩니다.

최근 동기화

체감 난이도

아직 평가 없음

첫 난이도 평가를 남겨주세요.

문제 추천

아직 평가 없음

이 문제가 도움이 되었는지 알려주세요.

이 문제 평가하기

GitHub 로그인 후 난이도를 제출하고, 원하는 경우 추천 여부도 함께 남길 수 있습니다.

추천 여부 (선택)

평가하려면 GitHub로 로그인해 주세요.

평가는 자동으로 저장소 Discussion에 기록되므로 Discussion 화면을 직접 열 필요가 없습니다. 같은 문제를 다시 평가하면 기존 평가가 갱신됩니다.

DISCUSSION

댓글

GitHub 로그인 후 작성할 수 있으며 모든 댓글은 이 저장소의 Discussions에 보관됩니다.