Cod sursa(job #2264884)

Utilizator filicriFilip Crisan filicri Data 20 octombrie 2018 12:30:59
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.13 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
#define lim 250000
#define inf 20000
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
priority_queue <pair <int,int> > pq;
vector <pair <int,int> > muc[lim];
int n,m,a,b,c,dist[lim];
bool viz[lim];
int main()
{
    f>>n>>m;
    for (int i=1;i<=m;i++)
    {
        f>>a>>b>>c;
        muc[a].push_back({(-1)*c,b});
    }
    for (int i=2;i<=n;i++) dist[i]=inf;
    pq.push({0,1});
    while (!pq.empty())
    {
        int nod=pq.top().second;
        pq.pop();
        for (int i=0;i<muc[nod].size();i++)
        {
            if (viz[muc[nod][i].second]==0)
            {
                viz[muc[nod][i].second]==1;
                if (dist[nod]-muc[nod][i].first<dist[muc[nod][i].second])
                {
                    dist[muc[nod][i].second]=dist[nod]-muc[nod][i].first;
                    pq.push({dist[muc[nod][i].second],muc[nod][i].second});
                }
            }
        }
    }
    for (int i=2;i<=n;i++)
        g<<dist[i]<<" ";
    f.close();
    g.close();
    return 0;
}