Cod sursa(job #2700981)

Utilizator goblinupufosPopescu Traian goblinupufos Data 29 ianuarie 2021 15:15:14
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.26 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

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

const int marime = 50005;
const int oo = 2e9;

int d[marime];
bool f[marime];
vector<pair<int, int>> mat[marime];

struct comp {
    bool operator () (int x, int y) {
        return d[x] > d[y];
    }
};

priority_queue<int, vector<int>, comp> q;

int main() {
    int n, m;
    fin >> n >> m;

    for (int i = 1; i <= m; ++i) {
        int x, y, cost;
        fin >> x >> y >> cost;
        mat[x].push_back({y, cost});
    }

    for (int i = 2; i <= n; ++i) {
        d[i] = oo;
    }

    q.push(1);
    f[1] = 1;

    while (!q.empty()) {
        int p = q.top();
        q.pop();
        f[p] = 0;

        for (int i = 0; i < mat[p].size(); ++i) {
            int vecin = mat[p][i].first;
            int cost = mat[p][i].second;

            if (d[vecin] > cost + d[p]) {
                d[vecin] = cost + d[p];
                if (f[vecin] == 0) {
                    q.push(vecin);
                    f[vecin] = 1;
                }
            }
        }
    }

    for (int i = 2; i <= n; ++i) {
        if (d[i] != oo)
            fout << d[i] << " ";
        else
            fout << "0 ";
    }

    return 0;
}