Cod sursa(job #3138743)

Utilizator chiriacandrei25Chiriac Andrei chiriacandrei25 Data 21 iunie 2023 20:29:42
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.91 kb
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

const int Nmax = 50005;

int n, m, dist[Nmax];
vector<pair<int, int>> L[Nmax];
priority_queue<pair<int, int>> Q;

void Dijkstra() {
    Q.push({0, 1});
    while(!Q.empty()) {
        int nod = Q.top().second;
        Q.pop();
        for(pair<int, int> x : L[nod]) {
            int vec = x.second;
            if(!dist[vec] || dist[vec] > dist[nod] + x.first) {
                dist[vec] = dist[nod] + x.first;
                Q.push({-dist[vec], vec});
            }
        }
    }
}

int main() {
    ifstream fin("dijkstra.in");
    ofstream fout("dijkstra.out");
    fin >> n >> m;
    for(int i = 1; i <= m; i++) {
        int x, y, z;
        fin >> x >> y >> z;
        L[x].push_back({y, z});
    }
    Dijkstra();
    for(int i = 2; i <= n; i++) {
        fout << dist[i] << " ";
    }
    return 0;
}