Cod sursa(job #2568945)

Utilizator FrostfireMagirescu Tudor Frostfire Data 4 martie 2020 10:32:41
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#define NMAX 50000
#define inf 2000000000

using namespace std;

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

int n, m, dist[NMAX+10];
bool viz[NMAX+10];
vector <pair <int, int> > nod[NMAX+10];
priority_queue <pair <int, int>, vector <pair <int, int> >, greater <pair <int, int> > > pq;

int main()
{
    f >> n >> m;
    for(int i=1; i<=m; i++)
        {   int nod1, nod2, cost;
            f >> nod1 >> nod2 >> cost;
            nod[nod1].push_back(make_pair(cost, nod2));
        }
    for(int i=2; i<=n; i++) dist[i] = inf;
    pq.push(make_pair(0, 1));
    while(!pq.empty())
        {   pair <int, int> a = pq.top();
            pq.pop();
            if(viz[a.second]) continue;
            viz[a.second] = 1;
            for(int i=0; i<nod[a.second].size(); i++)
                {   int d = nod[a.second][i].first + dist[a.second];
                    if(d < dist[nod[a.second][i].second])
                        {   dist[nod[a.second][i].second] = d;
                            pq.push(make_pair(d, nod[a.second][i].second));
                        }
                }
        }
    for(int i=2; i<=n; i++)
        {   if(dist[i] == inf) g << 0 << ' ';
            else g << dist[i] << ' ';
        }
    g << '\n';
    return 0;
}