Cod sursa(job #2983476)

Utilizator vladvoicux64Voicu Ioan Vladut vladvoicux64 Data 22 februarie 2023 15:27:49
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.54 kb
#include <fstream>
#include <vector>
#include <set>

using namespace std;

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

int N, M;
const int inf = 0x7fffffff;
vector<pair<int,int>> arcs[50001];
vector<int> distances;

int main() {
    in >> N >> M; //citim nr noduri/arce
    distances.resize(N+1, inf); //initial consideram distantele spre toate nodurile iufinite
    
    for (int i = 1; i <= M; ++i) {
        int cost, a, b;
        in >> a >> b >> cost; //citim arcele
        arcs[a].emplace_back(b, cost);
    }
    
    distances[1] = 0; //nodul de start
    
    set<int> tobevisited; //ce noduri sa vizitam
    tobevisited.insert(1); //incepem de la nodul de start
    
    while(!tobevisited.empty()){//cat timp avem ce vizita
        int nod = *tobevisited.begin();
        tobevisited.erase(tobevisited.begin());
        for(auto& vecin : arcs[nod]) {//verificam toate nodurile in care putem ajunge din nod
            int destinatie = vecin.first, cost = vecin.second;
            if(distances[destinatie] > distances[nod] + cost) {//daca putem ajunge intr-un nod i, si ajugnem mai rapid decat stiam pana acum
                distances[destinatie] = distances[nod] + cost;//actualizam distanta
                tobevisited.insert(destinatie);//si revizitam nodul sa verificam daca se scurteaza si alte distante
            }
        }
    }
    
    for (int i = 2; i <= N; ++i) {
        if (distances[i] == inf) {
            distances[i] = 0;
        }
        out << distances[i] << ' ';
    }
    out << '\n';
    
    in.close();
    out.close();
}