Cod sursa(job #2164508)

Utilizator CoCo1006Tudor Costin CoCo1006 Data 13 martie 2018 01:16:44
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.47 kb
#include <fstream>
#include <queue>
using namespace std;
ifstream in("dij.in");
ofstream out("dij.out");

#define Nmax 50001
const int oo = (1<<31) -1;

int N,M;
int dist[Nmax];
bool InCoada[Nmax];

struct compara{
    bool operator()(int x,int y)
    {
        return dist[x]>dist[y];
    }
};

priority_queue <int, vector <int>, compara> coada;

vector < pair <int, int> > Vecini[Nmax];

void citeste(){
   in>>N>>M;
   for(int i=1;i<=M;i++)
   {
       int x,y,c;
       in>>x>>y>>c;
       Vecini[x].push_back(make_pair(y,c));
   }
}

void afisare(){
    for(int i=2;i<=N;i++)
    {
        if(dist[i]==oo) out<<"0 ";
        else out<<dist[i]<<' ';
    }
}

void dij(int nod_plecare)
{
    for(int i=1;i<=N;i++) dist[i]=oo;

    dist[nod_plecare]=0;
    coada.push(nod_plecare);
    InCoada[nod_plecare] = 1;

    while(!coada.empty()){
        int nod_curent = coada.top();
        coada.pop();
        int l = Vecini[nod_curent].size();
        for(int i=0;i<l;i++)
        {
             int Vecin = Vecini[nod_curent][i].first;
             int Cost = Vecini[nod_curent][i].second;
             if(InCoada[Vecin]==0) {
                InCoada[Vecin] = 1;
                coada.push(Vecin);
             }
             if(dist[nod_curent] + Cost < dist[Vecin]){
                dist[Vecin] = dist[nod_curent] + Cost;
             }
        }
    }
}

int main()
{
    citeste();
    dij(1);
    afisare();
    return 0;
}