Cod sursa(job #3125874)

Utilizator vladp1324Vlad Pasare vladp1324 Data 4 mai 2023 18:28:31
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.26 kb
#include <fstream>
#include <vector>
#include <queue>
#define NMAX 50000

using namespace std;

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

int n, m, d[NMAX + 1];
bool ver[NMAX + 1];
const int INF = 2e9;
vector < pair < int, int > > v[NMAX + 1];

void dijkstra(int s) {
    for (int i = 1; i <= n; i++)
        d[i] = INF;
    priority_queue < pair <int, int> > pq;
    d[s] = 0;
    pq.push({0, s});
    while(!pq.empty()) {
        pair<int, int> f = pq.top();
        int cnc = f.first;
        int nc = f.second;
        pq.pop();
        if (!ver[nc]){
            for (int j = 0; j < v[nc].size(); j++) {
                int nv = v[nc][j].second;
                int cnv = v[nc][j].first;
                if (cnv + d[nc] < d[nv]) {
                    d[nv] = cnv + d[nc];
                    pq.push({-d[nv], nv});
                }
            }
            ver[nc] = true;
        }
    }
}

int main()
{
    fin >> n >> m;
    for (int i = 1; i <= m; i++) {
        int x, y, c;
        fin >> x >> y >> c;
        v[x].push_back({c, y});
    }
    dijkstra(1);
    for (int i = 2; i <= n; i++)
        if (d[i] == INF)
            fout << "0 ";
        else
            fout << d[i] << ' ';
    return 0;
}