Cod sursa(job #2207802)

Utilizator VNohaiNohai Vlad-Auras VNohai Data 26 mai 2018 20:36:41
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
using namespace std;
ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
int n, m;
const int NMAX=50005;
const int oo=(1<<30);
int d[NMAX];
bool incoada[NMAX];

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

priority_queue < int, vector < int >, comp > q;

vector < pair <int,int> > G[NMAX];



void dij(int poz)
{
   for(int i=2; i<=n; i++)
   d[i]=oo;
   d[poz]=0;
   q.push(poz);
   incoada[poz]=true;
   while(!q.empty())
   {
   int nod=q.top();
   q.pop();
   incoada[nod]=false;
   for(int i=0; i<G[nod].size(); i++)
   {
    int vecin=G[nod][i].first;
    int cost=G[nod][i].second;
    if(d[nod]+cost<d[vecin])
    {
    d[vecin]=d[nod]+cost;
    if(incoada[vecin]==false)
      {
      q.push(vecin);
      incoada[vecin]=true;
      }
    }
   }
   }
}

int main()
{
    int a, b, c;
    f>>n>>m;
    for(int i=1; i<=m; i++)
    {
    f>>a>>b>>c;
    G[a].push_back(make_pair(b, c));
    }
    dij(1);
    for(int i=2; i<=n; i++)
    if(d[i]!=oo)
    g<<d[i]<<" ";
    else
    g<<'0';
    return 0;
}