Cod sursa(job #2886853)

Utilizator hobbitczxdumnezEU hobbitczx Data 8 aprilie 2022 14:35:58
Problema Algoritmul lui Dijkstra Scor 80
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include <bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;

const string fisier = "dijkstra";

ifstream fin (fisier + ".in");
ofstream fout (fisier + ".out");

const int N_MAX = 5e4 + 5;

set<pair<int , int>>s;
vector<pair<int , int>>g[N_MAX];
int d[N_MAX] , n , m;

int main(){
    ios_base::sync_with_stdio(false);
    fin >> n >> m;
    for (int i=1; i<=m; i++){
        int x , y , cost; fin >> x >> y >> cost;
        g[x].push_back({y , cost});
    }
    fill(d + 1 , d + n + 1 , INF);
    d[1] = 0;
    s.insert({1 , 0});
    while (s.empty() == false){
        pair<int , int>p = *(s.begin());
        s.erase(s.begin());
        ll costActual = p.second , node = p.first;
        if (d[node] < costActual) continue;
        for (int i=0; i<(int)g[node].size(); i++){
            int new_node = g[node][i].first , new_cost = g[node][i].second;
            if (d[new_node] > d[node] + new_cost){
                d[new_node] = d[node] + new_cost;
                s.insert({new_node , d[new_node]});
            }
        }
    }
    for (int i=2; i<=n; i++){
        fout << ((d[i] == INF) ? 0 : d[i]) << " ";
    }
}