Cod sursa(job #3336656)

Utilizator iulia_learning_timeLearning Time iulia_learning_time Data 25 ianuarie 2026 11:39:39
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.19 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

// const int INF = 1e9;
const int INF = 2000000000;

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;
}