Cod sursa(job #2886861)

Utilizator hobbitczxdumnezEU hobbitczx Data 8 aprilie 2022 14:54:19
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.21 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 = 50005;

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});
    }
    memset(d , INF , sizeof(d));
    d[1] = 0;
    s.insert({0 , 1});
    while (s.empty() == false){
        int node = s.begin() -> second;
        s.erase(s.begin());
        for (auto it = g[node].begin(); it != g[node].end(); ++it){
            int new_node = it -> first , cost = it -> second;
            if (d[new_node] > d[node] + cost){
                if (d[new_node] != INF){
                    s.erase(s.find(make_pair(d[new_node] , new_node)));
                }
                d[new_node] = d[node] + cost;
                s.insert({d[new_node] , new_node});
            }
        }
    }
    for (int i=2; i<=n; i++){
        fout << ((d[i] == INF) ? 0 : d[i]) << " ";
    }
}