Cod sursa(job #1863455)

Utilizator tamionvTamio Vesa Nakajima tamionv Data 30 ianuarie 2017 21:58:57
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.93 kb
#include <bits/stdc++.h>
using namespace std;

constexpr int inf = 1e9 + 100, maxn = 5e4 + 100;

ifstream f("dijkstra.in");
ofstream g("dijkstra.out");
vector<pair<int, int>> vec[maxn] = {};
bitset<maxn> viz = 0;
priority_queue<tuple<int, int>, vector<tuple<int, int>>, greater<tuple<int, int>>>
    pq;
int dist[maxn] = {}, n, m;

void djk(){
    fill(dist, dist+maxn, inf);
    pq.emplace(0, 0);
    for(int cur, cur_dist; !pq.empty(); ){
        tie(cur_dist, cur) = pq.top();
        pq.pop();
        if(dist[cur] <= cur_dist) continue;
        dist[cur] = cur_dist;
        for(const auto next : vec[cur]){
            pq.emplace(cur_dist + next.first, next.second); } } }

int main(){
    f >> n >> m;
    for(int i = 0, x, y, c; i < m; ++i){
        f >> x >> y >> c;
        --x, --y;
        vec[x].emplace_back(c, y); }
    djk();
    replace(dist, dist+maxn, inf, 0);
    for(int i = 1; i < n; ++i) g << dist[i] << ' ';
    return 0; }