Cod sursa(job #2420995)

Utilizator ela_topaTopa Elena ela_topa Data 13 mai 2019 19:15:00
Problema Algoritmul lui Dijkstra Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.2 kb
#include<bits/stdc++.h>
using namespace std;

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

constexpr int NX = 50010;
constexpr int inf = 99999999;
int N, M;
int viz[NX];
stack <int> S;
int dist[NX];
vector <pair <int, int>> V[NX];
priority_queue < pair <int, int> > pq;

int main()
{
    fin>>N>>M; //citesc nr de noduri si muchiile
    int x, y, c;
    for(int i=1; i<=M; ++i)
    {
        fin>>x>>y>>c;
        V[x].push_back( {c, y} ); //le bag in lista de vecini
    }
    //iau nodul 1 ca sursa
    dist[1] = 0; //ii fac dist 0
    pq.push( {0,1} ); //bag in pq nodul 1 cu costul 0

    for(int i=2; i<=N; ++i)
    {
        dist[i] = inf;
    }

    while(!pq.empty())
    {
        int nod = pq.top().second;
        pq.pop(); //scot
        for(auto &i: V[nod])
        {
            int oDist = dist[nod] + i.first; //distanta nodului plus cea a vecinului
            if(oDist < dist[i.second]) //optimizez
            {
                dist[i.second] = oDist;

                pq.push({oDist,i.second});
            }
        }

    }

    for(int i=2; i<=N; ++i) //afisez
    {
        if(dist[i]==inf)
            fout << 0 <<" ";
        else
            fout<<dist[i]<<" ";
    }
    return 0;
}