Cod sursa(job #2514697)

Utilizator Catalin_GriuGriu Catalin Catalin_Griu Data 26 decembrie 2019 17:06:40
Problema Algoritmul lui Dijkstra Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.13 kb
#include <bits/stdc++.h>
#define NMAX 50005
using namespace std;

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

int n, m, s, l, r;
set < pair<int, int> > v[NMAX];
int d[NMAX];

void dijkstra(int nod)
{
    priority_queue <pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>> > h;
    memset(d, -1, NMAX);
    d[nod] = 0;
    bool vizitat[n+3];
    memset(vizitat, false, n+3);
    h.push(make_pair(0, nod));

    while(!h.empty())
    {
        auto i = h.top();
        h.pop();
        nod = i.second;
        vizitat[nod] = true;
        for(auto j: v[nod])
        {
            if(!vizitat[j.first] && (d[j.first]==-1 || d[j.first]>d[nod]+ j.second))
            {
                d[j.first] = d[nod]+ j.second;
                h.push(make_pair(d[j.first], j.first));
            }
        }

    }
}

int main()
{
    fin>>n>>m;
    int x, y, w;
    while(m--)
    {
        fin>>x>>y>>w;
        v[x].insert(make_pair(y, w));
    }
    dijkstra(1);
    for(int i=2; i<=n; i++)
        if(d[i]!=-1)
            fout<<d[i]<< ' ';
        else fout<<0<<' ';

    return 0;
}