Pagini recente » Cod sursa (job #856553) | Cod sursa (job #1015986) | Cod sursa (job #139896) | Cod sursa (job #327233) | Cod sursa (job #3268559)
#include <iostream>
#include <vector>
#include <queue>
#include <fstream>
#include <climits>
using namespace std;
void dijkstra(int N, const vector<vector<pair<int, int>>>& graf, vector<int>& distante) {
// Heap pentru perechi (distanta, nod), ordonate crescător după distanță
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> heap;
distante[1] = 0; // Distanța de la nodul 1 la el însuși este 0
heap.push({0, 1}); // Adăugăm nodul 1 în heap cu distanța 0
while (!heap.empty()) {
int distU = heap.top().first;
int u = heap.top().second;
heap.pop();
// Dacă distanța extrasă din heap este mai mare decât cea deja calculată, o ignorăm
if (distU > distante[u])
continue;
// Relaxăm muchiile care pleacă din u
for (auto [v, cost] : graf[u]) {
if (distante[u] + cost < distante[v]) {
distante[v] = distante[u] + cost;
heap.push({distante[v], v});
}
}
}
}
int main() {
ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");
int N, M;
fin >> N >> M;
vector<vector<pair<int, int>>> graf(N + 1); // Graful orientat, indexat de la 1
vector<int> distante(N + 1, INT_MAX); // Vector pentru distanțele minime
for (int i = 0; i < M; i++) {
int A, B, C;
fin >> A >> B >> C;
graf[A].push_back({B, C}); // Arc de la A la B cu costul C
}
// Apelăm Dijkstra pornind de la nodul 1
dijkstra(N, graf, distante);
// Scriem rezultatele în fișier
for (int i = 2; i <= N; i++) {
if (distante[i] == INT_MAX)
fout << "-1 "; // Dacă nu există drum, afișăm -1
else
fout << distante[i] << " ";
}
return 0;
}