Cod sursa(job #1918276)

Utilizator eu3neuomManghiuc Teodor-Florin eu3neuom Data 9 martie 2017 14:48:35
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1.15 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];

struct cmp {
    bool operator()(const int &a, const int &b) {
        return D[a] > D[b];
    }
};

inline void Dijkstra() {
    priority_queue < int, vector < int >, cmp > pq;
    pq.push(1);

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

        for(auto const &it: G[node]) {
            if(D[node] + it.second < D[it.first]) {
                D[it.first] = D[node] + it.second;
                pq.push(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;
}