Pagini recente » Cod sursa (job #1978039) | Cod sursa (job #3292652) | Cod sursa (job #44531) | Cod sursa (job #544595) | Cod sursa (job #2857451)
#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
using namespace std;
//Sa se determine lungimea minima a drumului de la nodul 1 la fiecare din nodurile 2, 3, ..., N-1, N si sa se afiseze aceste lungimi.
//Lungimea unui drum este data de suma lungimilor arcelor care constituie drumul.
//Lungimea nu poate fi negativa (si nici nula)
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
const int MAX = 50005, INF = (1 << 30);
int n, m;
int distanta[MAX];
bool inclusInHeap[MAX];
vector < pair < int, int > > legaturi[MAX];
struct Compara //criteriu de ordonare crescatoare a heap-ului, in functie de costul nodului
{
bool operator()(int x, int y)
{
return (distanta[x] > distanta[y]);
}
};
priority_queue < int, vector < int >, Compara > heap;
void Citire()
{
fin >> n >> m;
for (int i = 1; i <= m; i++)
{
int x, y, cost;
fin >> x >> y >> cost;
legaturi[x].push_back(make_pair(y, cost));
}
}
void Dijkstra (int nodStart)
{
for (int i = 1; i <= n; i++) //costul tuturor nodurilor devine infinit
{
distanta[i] = INF;
}
distanta[nodStart] = 0; //costul nodului de plecare devine 0
heap.push(nodStart); //includem nodul de plecare in heap
inclusInHeap[nodStart] = true;
while(!heap.empty()) //cat timp avem noduri in heap
{
int nodCurent = heap.top(); //extragem cel mai ieftin nod din heap
heap.pop();
inclusInHeap[nodCurent] = false;
for (unsigned int i = 0; i < legaturi[nodCurent].size(); i++) //ii parcurgem vecinii
{
int vecin = legaturi[nodCurent][i].first;
int cost = legaturi[nodCurent][i].second;
if (distanta[nodCurent] + cost < distanta[vecin]) //actualizam costul vecinului daca e posibil
{
distanta[vecin] = distanta[nodCurent] + cost;
if (inclusInHeap[vecin] == false) //includem vecinul in heap daca nu e deja inclus
{
heap.push(vecin);
inclusInHeap[vecin] = true;
}
}
}
}
}
void Afisare()
{
for (int i = 2; i <= n; i++)
{
if (distanta[i] != INF)
{
fout << distanta[i] << ' ';
}
else
{
fout << "0 ";
}
}
}
int main()
{
Citire();
Dijkstra(1);
Afisare();
return 0;
}