Cod sursa(job #1918301)

Utilizator eu3neuomManghiuc Teodor-Florin eu3neuom Data 9 martie 2017 14:54:39
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.16 kb
#include <bits/stdc++.h>

using namespace std;

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

const int NMax = 5e4 + 5;
const int INF = 1e9;

vector < pair < int, int > > G[NMax];
int D[NMax];

inline void Dijkstra() {
    priority_queue < pair < int, int >, vector < pair < int, int > >, greater < pair < int, int > > > pq;
    pq.push({0, 1});

    while(!pq.empty()) {
        int node = pq.top().second;
        int value = pq.top().first;
        pq.pop();

        for(auto const &it: G[node]) {
            if(value + it.second < D[it.first]) {
                D[it.first] = value + it.second;
                pq.push({D[it.first], it.first});
            }
        }
    }
}

int main(){
    ios::sync_with_stdio(false);

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

    for(int i = 1; i <= m; i++) {
        int a, b, c;
        fin >> a >> b >> c;

        G[a].push_back({b, c});
    }

    for(int i = 2; i <= n; i++) D[i] = INF;
    Dijkstra();

    for(int i = 2; i <= n; i++) {
        if(D[i] == INF) {
            fout << "0 ";
        } else {
            fout << D[i] << " ";
        }
    }

    return 0;
}