Cod sursa(job #3158821)

Utilizator xxUnrealUxxNiculae Adrian-Ioan xxUnrealUxx Data 19 octombrie 2023 21:06:24
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.03 kb
#include <fstream>
#include <vector>
#include <queue>
#include <climits>

using namespace std;

const int Nmax = 50005;

int n, m;
vector<int> dist(Nmax, INT_MAX);
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, dist_nod = -Q.top().first;
        Q.pop();
        if(dist_nod > dist[nod]) {
            continue;
        }
        for(pair<int, int> x : L[nod]) {
            int vec = x.first;
            if(dist[vec] > dist[nod] + x.second) {
                dist[vec] = dist[nod] + x.second;
                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;
}