Cod sursa(job #3268559)

Utilizator Anamaria240Rusu Ana-Maria Anamaria240 Data 15 ianuarie 2025 22:44:03
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.85 kb
#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;
}