LeetCode #1976

Number of Ways to Arrive at Destination

1개의 풀이 · C++

문제 원문 보기 ↗

SOLUTION INFO

C++ · main.cpp

main.cpp
using ll = long long;
class Solution {
public:
    int countPaths(int N, vector<vector<int>>& roads) {
        vector<vector<pair<int, ll>>> G(N);
        for(int i = 0; i < (int)roads.size(); ++i) {
            int u = roads[i][0], v = roads[i][1];
            ll w = roads[i][2];
            G[u].emplace_back(v, w);
            G[v].emplace_back(u, w);
        }
        const ll INF = LLONG_MAX / 2;
        priority_queue<pair<ll, int>> pq;
        vector<ll> dist(N, INF), DP(N);
        dist[0] = 0; DP[0] = 1; pq.emplace(0, 0);
        const ll MOD = 1e9 + 7;
        while(!pq.empty()) {
            auto [d, cur] = pq.top(); pq.pop();
            if(dist[cur] != -d) continue;
            for(auto [nxt, w]: G[cur]) {
                if(dist[nxt] > dist[cur] + w) {
                    dist[nxt] = dist[cur] + w;
                    DP[nxt] = DP[cur];
                    pq.emplace(-dist[nxt], nxt);
                }
                else if(dist[nxt] == dist[cur] + w) DP[nxt] = (DP[nxt] + DP[cur]) % MOD;
            }
        }
        return DP[N - 1];
    }
};

SOLUTION DESCRIPTION

풀이 설명

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