Cod sursa(job #2765386)

Utilizator florinrafiliuRafiliu Florin florinrafiliu Data 26 iulie 2021 18:12:57
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.16 kb
#include <iostream>
#include <fstream>
#include <set>
#include <vector>
using namespace std;

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

const int INF = 0x3f3f3f3f;
vector <pair<int,int> > g[50005];
int dist[50005];


int main()
{
    int n, m; fin >> n >> m;

    for(int i = 1; i <= m; ++i) {
        int u, v, c; fin >> u >> v >> c;
        g[u].push_back(make_pair(v,c));
    }

    for(int i = 1; i <= n; ++i)
        dist[i] = INF;
    dist[1] = 0;

    set <pair<int,int> > d;
    d.insert(make_pair(0, 1));

    while(!d.empty()) {
        int node = d.begin()->second;

        d.erase(d.begin());

        for(auto v : g[node]) {
            int nod = v.first;
            int cost = v.second;
            if(dist[nod] > dist[node] + cost) {
                if(dist[nod] != INF)
                    d.erase(d.find(make_pair(dist[nod], nod)));
                dist[nod] = dist[node] + cost;
                d.insert(make_pair(dist[nod], nod));
            }
        }
    }

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

    return 0;
}