Cod sursa(job #2682155)

Utilizator andrei.florea0405Florea Andrei-Bogdan andrei.florea0405 Data 7 decembrie 2020 22:04:39
Problema Algoritmul lui Dijkstra Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.4 kb
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define fi first
#define se second
#define MOD 1000000007

typedef vector<int> vi;
typedef vector<vi> vvi;
typedef pair<int, int> ii;
typedef vector<ii> vii;
typedef long long ll;
typedef vector<ll> vll;
typedef vector<vll> vvll;
typedef double ld;

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

int n, m;
vii G[50005];
int dist[50005];


int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    
    fin >> n >> m;
    for (int i = 1; i <= m; i++) {
        int x, y, c;
        fin >> x >> y >> c;
        G[x].pb(make_pair(y, c));
    }
    
    for (int i = 1; i <= n; i++) {
        dist[i] = INT_MAX;
    }

    vii heap;
    heap.pb(make_pair(0, 1));
    push_heap(heap.begin(), heap.end());
    while (!heap.empty()) {
        pop_heap(heap.begin(), heap.end());
        int d = heap.back().first;
        int node = heap.back().second;
        heap.pop_back();

        for (auto &vecin : G[node]) {
            if (dist[vecin.first] > d + vecin.second) {
                dist[vecin.first] = d + vecin.second;
                heap.push_back(make_pair(dist[vecin.first], vecin.first));
                push_heap(heap.begin(), heap.end());
            }
        }
    }
    
    for (int i = 2; i <= n; i++) {
        if (dist[i] == INT_MAX) {
            dist[i] = 0;
        }
        fout << dist[i] << " ";
    }

    return 0;
}