Cod sursa(job #2632811)

Utilizator k2e0e0w3qDumitrescu Gheorghe k2e0e0w3q Data 5 iulie 2020 02:41:38
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <bits/stdc++.h>
#define INF 0x3F3F3F3F
#define N 50000
using namespace std;

int dist[N+1];
vector <pair <int, int>> G[N+1];

int main () {
    ifstream fin ("dijkstra.in");
    ofstream fout ("dijkstra.out");

    int n, m;
    fin >> n >> m;

    int i, j, c;
    for (; m; m--) {
        fin >> i >> j >> c;
        G[i].push_back({c, j});
    }

    fill(&dist[2], &dist[n+1], INF);
    priority_queue <pair <int, int>, vector <pair <int, int>>, greater <pair <int, int>>> PQ;
    PQ.push({0, 1});

    while (!PQ.empty()) {
        auto save=PQ.top();
        PQ.pop();

        if (dist[save.second] == save.first)
            for (auto it: G[save.second])
                if (dist[it.second] > save.first + it.first)
                    dist[it.second] = save.first + it.first,
                    PQ.push({dist[it.second], it.second});
    }

    for (i=2; i<=n; ++i)
        fout << (dist[i]==INF?0:dist[i]) << ' ';
    return 0;
}