Cod sursa(job #2815798)

Utilizator Robert.BrindeaBrindea Robert Robert.Brindea Data 10 decembrie 2021 12:48:42
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.05 kb
#include <bits/stdc++.h>

#define MAXN 50007
#define INF 20010

using namespace std;

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

int n, m, d[MAXN];
vector<pair<int, int> > adj[MAXN];
set<pair<int, int> > q;

void dij()
{
    memset(d, INF, sizeof(d));
    d[0] = d[1] = 0;
    q.insert({0, 1});
    while(!q.empty())
    {
        pair<int, int> crt = *q.begin();
        q.erase(q.begin());
        for(auto& it: adj[crt.second])
        {
            if(d[it.second] > d[crt.second] + it.first)
            {
                if(d[it.second] != INF) q.erase(q.find({d[it.second], it.second}));
                d[it.second] = d[crt.second] + it.first;
                q.insert({d[it.second], it.second});
            }
        }
    }
}

int main()
{
    fin >> n >> m;
    for(int i = 0; i < m; i++)
    {
        int x, y, c;
        fin >> x >> y >> c;
        adj[x].push_back({c, y});
    }
    dij();
    for(int i = 2; i <= n; i++)
        fout << (d[i]==INF? 0 : d[i]) << " ";

    return 0;
}