Cod sursa(job #3159569)

Utilizator iusty64Iustin Epanu iusty64 Data 21 octombrie 2023 16:59:07
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.02 kb
#include <iostream>
#include <fstream>
#include <queue>
#include <climits>
using namespace std;

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

int const Vmax = 50001;
priority_queue<pair<int, int>> Q;
vector<pair<int, int>> v[Vmax];
vector<int> dist(Vmax, INT_MAX);

void bfs(int distanta, int nod){
    dist[nod] = distanta;
    Q.push({dist[nod], nod});
    while(!Q.empty()){
        int vec = Q.top().second;
        int distanta_vec = -Q.top().first;
        if(distanta_vec > dist[vec])
            continue;
        Q.pop();
        for(auto i : v[vec]){
            if(dist[i.second] > dist[vec] + i.first){
                dist[i.second] = dist[vec] + i.first;
                Q.push({-dist[i.second], i.second});
            }
        }
    }
}

int main()
{
    int n, m, z;
    fin>>n>>m;
    while(m--){
        int x, y;
        fin>>x>>y>>z;
        v[x].push_back({z, y});
    }
    bfs(0, 1);
    for(int i=2;i<=n;i++)
        fout<<dist[i]<<" ";
    return 0;
}