Cod sursa(job #3268159)

Utilizator divadddDavid Curca divaddd Data 13 ianuarie 2025 19:55:10
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.9 kb
#include <bits/stdc++.h>
using namespace std;
const int NMAX = 5e4+2;
const int INF = 1e9;
using pii = pair<int, int>;
int n,m,dist[NMAX];
vector<pii> v[NMAX];

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

int main() {
  fin >> n >> m;
  for(int i = 1; i <= m; i++){
    int x, y, z;
    fin >> x >> y >> z;
    v[x].push_back({y, z});
    // v[y].push_back({x, z});
  }
  for(int i = 1; i <= n; i++){
    dist[i] = INF;
  }
  set<pii> s;
  dist[1] = 0;
  s.insert({0, 1});
  while(!s.empty()){
    auto [accum, nod] = *s.begin();
    s.erase(s.begin());
    for(auto [vecin, cost]: v[nod]){
      if(dist[vecin] > accum + cost){
        s.erase({dist[vecin], vecin});
        dist[vecin] = accum + cost;
        s.insert({dist[vecin], vecin});
      }
    }
  }
  for(int i = 2; i <= n; i++){
    fout << (dist[i] == INF ? 0 : dist[i]) << " ";
  }
  return 0;
}