Cod sursa(job #1933072)

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

using namespace std;

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


bool cmp(int a, int b) {
   return d[a] > d[b];
}

priority_queue<int, vector<int>, function<bool(int, int)> > q(cmp);

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(1);
    viz[1] = true;

    while(!q.empty()) {
        int nod = q.top(); q.pop();
        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(j.first);
                }
        }
    }

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