Cod sursa(job #2495542)

Utilizator deiubejanAndrei Bejan deiubejan Data 19 noiembrie 2019 16:49:54
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.21 kb
#include <bits/stdc++.h>
using namespace std;

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

#define cin fin
#define cout fout
/*
*/

const int NMAX=5e4+10;
const int oo=1e9;
int n, m, x, y, c;
vector<pair<int, int>>rel[NMAX];
set<pair<int, int>>sord;
int d[NMAX];
int t[NMAX];

void read()
{
    cin>>n>>m;
    for(int i=1; i<=m; i++)
    {
        cin>>x>>y>>c;
        rel[x].push_back({y, c});
    }
}


void solve()
{
    for(int i=2; i<=n; i++)
        d[i]=oo;
    sord.insert({0, 1});
    while(!sord.empty())
    {
        int nod=sord.begin()->second;
        int dist=sord.begin()->first;
        sord.erase(sord.begin());
        for(auto el:rel[nod])
        {
            int to=el.first;
            int cost=el.second;
            if(d[to]>d[nod]+cost)
            {
                if(d[to]!=oo)
                    sord.erase({d[to], to});
                t[to]=nod;
                d[to]=d[nod]+cost;
                sord.insert({d[to], to});
            }
        }
    }
    for(int i=2; i<=n; i++)
    {
        if(d[i]==oo)
            d[i]=0;
        cout<<d[i]<<" ";
    }

}


int main()
{
    read();
    solve();

    return 0;
}