Cod sursa(job #2529739)

Utilizator PatrascuAdrian1Patrascu Adrian Octavian PatrascuAdrian1 Data 23 ianuarie 2020 21:40:46
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <bits/stdc++.h>

using namespace std;

typedef pair<int,int> pii;
priority_queue <pii,vector<pii>,greater<pii> > q;
const int Max = 2e9 + 5;
int N,M;

int main() {
    freopen("dijkstra.in","r",stdin);
    freopen("dijkstra.out","w",stdout);
    cin >> N >> M;
    vector<pii> a[N + 1];
    vector<bool> seen(N + 1,0);
    while(M--) {
        int x,y,c;
        cin >> x >> y >> c;
        a[x].push_back({c,y});
    }
    vector<int> dp(N + 1,Max);
    dp[1] = 0;
    for(auto i : a[1]) {
        q.push(i);
        dp[i.second] = i.first;
    }
    seen[1] = 1;

    while(!q.empty()) {
        while(!q.empty() && seen[q.top().second])
            q.pop();
        if(!q.empty()) {
            int node = q.top().second;
            seen[node] = 1;
            q.pop();
            for(auto i : a[node]) {
                int cost = i.first,newnode = i.second;
                if(dp[node] + cost < dp[newnode]) {
                    dp[newnode] = dp[node] + cost;
                    q.push({dp[newnode],newnode});
                }
            }
        }
    }
    for(int i = 2; i <= N; ++i) {
        (dp[i] == Max ? cout << 0 : cout << dp[i]);
        cout << " ";
    }
    return 0;
}