Pagini recente » Cod sursa (job #3349129) | Cod sursa (job #77129) | Cod sursa (job #974716) | Borderou de evaluare (job #1518760) | Cod sursa (job #3336652)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
const int INF = 1e9;
ifstream fin("dijkstra.in"); // Presupunem fisiere specifice Dijkstra
ofstream fout("dijkstra.out");
void Dijkstra(int startNode, int N, vector<vector<pair<int, int>>>& adj) {
// 1. Min-heap: { distanță_de_la_sursă, nod_curent }
// NU mai avem nevoie de părinte în coadă pentru calculul distanțelor
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> pq;
// 2. Vector de distanțe (în loc de visited)
vector<int> dist(N + 1, INF);
// Setăm startul
dist[startNode] = 0;
pq.push({0, startNode});
while (!pq.empty()) {
int d = pq.top().first; // Distanța totală până aici
int u = pq.top().second; // Nodul curent
pq.pop();
// Optimizare: Dacă am scos din coadă o distanță mai mare decât cea
// pe care o știm deja în dist[u], o ignorăm (e o intrare veche)
if (d > dist[u])
continue;
// Parcurgem vecinii
for (auto &edge : adj[u]) {
int v = edge.first;
int weight = edge.second;
// 3. RELAXAREA (Core-ul Dijkstra)
// Dacă drumul prin 'u' este mai scurt decât vechiul drum spre 'v'
if (dist[u] + weight < dist[v]) {
dist[v] = dist[u] + weight; // Actualizăm distanța
pq.push({dist[v], v}); // Punem noua distanță în coadă
}
}
}
// Afișare distanțe
for (int i = 2; i <= N; i++) {
if (dist[i] == INF)
fout << "0";
else
fout << dist[i] << " ";
}
}
int main() {
int N, M;
fin >> N >> M;
vector<vector<pair<int, int>>> adj(N + 1);
for (int i = 0; i < M; i++) {
int u, v, c;
fin >> u >> v >> c;
// Dijkstra e de obicei pe graf orientat, dar merge și pe neorientat
adj[u].push_back({v, c});
// adj[v].push_back({u, c}); // Decomentează dacă e neorientat
}
int startNode = 1; // De obicei se cere distanța de la nodul 1
Dijkstra(startNode, N, adj);
return 0;
}