Pagini recente » Cod sursa (job #2932871) | Cod sursa (job #3192746) | Cod sursa (job #270090) | Cod sursa (job #492831) | Cod sursa (job #2807506)
#include <bits/stdc++.h>
#define inf 1000000
#define Nmax 50005
using namespace std;
ifstream fin ("dijkstra.in");
ofstream fout ("dijkstra.out");
priority_queue <pair<int,int>> q; // coada cu prioritati {cost,nod}
vector <pair<int,int>> muchii[Nmax];
bool viz[Nmax]; // viz[x] = 1 daca nodul a fost vizitat
int N, M, dist[Nmax]; // nr noduri, nr muchii, dist[x] = distanta de la nodul de start la nodul x
void Read()
{
fin >> N >> M;
for(int i = 1; i <= M; ++i)
{
int x, y, cost;
fin >> x >> y >> cost;
muchii[x].push_back({cost,y});
}
}
void Dijkstra(int s)
{
// initial presupunem fiecare distanta ca fiind infinit
for(int i = 1; i <= N; ++i)
dist[i] = inf;
q.push({0,s}); // adaugam in coada nodul de inceput cu costul 0 (de la s la s avem distanta 0)
dist[s] = 0; // distanta de la s la s va fi 0
viz[s] = 1;
while(!q.empty())
{
int nod_curent = q.top().second; // nodul curent
q.pop();
viz[nod_curent] = 0;
for(auto i : muchii[nod_curent]) // parcurgem nodurile adiacente cu nodul curent si actualizam distantele minime
{
int y = i.second; // nodul adiacent cu nodul curent
int cost = i.first; // costul de la nodul curent pana la y
if(dist[nod_curent] + cost < dist[y])
{
dist[y] = dist[nod_curent] + cost;
if(!viz[y]) // marcam nodul ca fiind vizitat daca nu era si il adaugam in coada
{
viz[y] = 1;
q.push({dist[y],y});
}
}
}
}
}
void Afisare()
{
for(int i = 2; i <= N; ++i)
{
if(dist[i] != inf)
fout << dist[i] << " ";
else
fout << 0 << " ";
}
}
int main()
{
Read();
Dijkstra(1);
Afisare();
return 0;
}