Cod sursa(job #1933066)

Utilizator serbanSlincu Serban serban Data 20 martie 2017 13:26:59
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.05 kb
#include <bits/stdc++.h>
#define oo 2147483647

using namespace std;

vector< pair<int, int> > G[50005];
priority_queue< pair<int, int> > q;
int d[50005];
bool viz[50005];

int main()
{
    ifstream f("dijkstra.in");
    ofstream g("dijkstra.out");

    int n, m, x, y, cost;
    f >> n >> m;
    for(int i = 1; i <= m; i ++) {
        f >> x >> y >> cost;
        G[x].push_back({y, cost});
    }

    for(int i = 2; i <= n; i ++) d[i] = oo;

    q.push({0, 1});
    viz[1] = true;

    while(!q.empty()) {
        pair<int, int> act = q.top(); q.pop();
        int nod = act.second;
        int cost = d[nod];

        for(auto j: G[nod]) {
            if(d[j.first] > cost + j.second)
                d[j.first] = cost + j.second;
                if(!viz[j.first]) {
                    viz[j.first] = true;
                    q.push( {d[j.first], j.first} );
                }
        }
    }

    for(int i = 2; i <= n; i ++) {
        if(d[i] == oo) d[i] = 0;
        g << d[i] << " ";
    }
    g << "\n";
    return 0;
}