Cod sursa(job #3165056)

Utilizator octavian202Caracioni Octavian Luca octavian202 Data 5 noiembrie 2023 12:49:06
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.26 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>

using namespace std;

ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

const int NMAX = 50003, INF = 1e8;
vector<pair<int, int> > g[NMAX]; // nod, cost
int d[NMAX], n;

struct dcmp {
    bool operator()(const pair<int, int> &a, const pair<int, int> &b) {
        return a.second < b.second;
    }
};

void dijkstra(int s) {
    for (int i = 1; i <= n; i++)
        d[i] = INF;
    d[s] = 0;

    priority_queue<pair<int, int>, vector<pair<int, int> >, dcmp> pq;
    for (auto nxt : g[s]) {
        pq.push(nxt);
    }

    while (!pq.empty()) {
        int nod = pq.top().first, cost = pq.top().second;
        pq.pop();

        if (cost > d[nod])
            return;

        d[nod] = cost;
        for (auto nxt : g[nod]) {
            if (cost + nxt.second < d[nxt.first]) {
                d[nxt.first] = cost + nxt.second;
                pq.push(make_pair(nxt.first, cost + nxt.second));
            }
        }
    }
}

int main() {

    int m;
    fin >> n >> m;
    while (m--) {
        int x, y, c;
        fin >> x >> y >> c;
        g[x].push_back(make_pair(y, c));
    }

    dijkstra(1);
    for (int i = 1; i <= n; i++)
        fout << d[i] << ' ';


    return 0;
}