Cod sursa(job #3206890)

Utilizator ggaaggaabbiigoteciuc gabriel ggaaggaabbii Data 24 februarie 2024 13:24:35
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.18 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <climits>

using namespace std;

#define MAXN 50010

ifstream f("dijkstra.in");
ofstream g("dijkstra.out");

vector<pair<int, int>> G[MAXN];
priority_queue<pair<int, int>> PQ;

int n, m, x, y, c;
int costuri[MAXN];

void dijkstra(int nod) {
    for (int i = 1; i <= n; ++i) {
        costuri[i] = INT_MAX;
    }
    costuri[nod] = 0;
    PQ.push({0, nod});
    while (PQ.size()) {
        auto aux = PQ.top();
        int nodCurrent = aux.second;
        PQ.pop();

        for (auto it:G[nodCurrent]) {
            if (costuri[nodCurrent] + it.second < costuri[it.first]) {
                costuri[it.first] = costuri[nodCurrent] + it.second;
                PQ.push({-(costuri[nodCurrent] + it.second), it.first});
            }
        }
    }
}

int main()
{
    f >> n >> m;

    while (m--) {
        f >> x >> y >> c;
        G[x].push_back({y, c});
    }

    dijkstra(1);

    for (int i = 2; i <= n; ++i) {
        if (costuri[i] == INT_MAX) {
            g << "0 ";
        } else {
            g << costuri[i] << ' ';
        }
    }

    return 0;
}