Cod sursa(job #3256819)

Utilizator darian4444Verniceanu Darian darian4444 Data 16 noiembrie 2024 10:25:22
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <bits/stdc++.h>

using namespace std;

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

const int inf = 1e9;

int N,M,i,j,F[50005],d[50005],a,b,c;

vector< pair<int,int> > A[50005];

priority_queue< pair<int,int> > pq;

int main()
{
    fin >> N >> M;

    for (i = 1;i <= N;i++){
        fin >> a >> b >> c;
        A[a].push_back({b,c});
    }

    for (i = 2;i <= N;i++)
        d[i] = inf;

    pq.push({0,1});

    while (!pq.empty()){
        b = pq.top().second;
        pq.pop();

        if (F[b] == 1)
            continue;

        for (auto nextn : A[b]){
            if (d[nextn.first] > d[b] + nextn.second){
                d[nextn.first] = d[b] + nextn.second;
                pq.push({-nextn.second,nextn.first});
            }
        }

        F[b] = 1;
    }

    for (i = 2;i <= N;i++)
        if (d[i] == inf)
            fout << "0 ";
        else
            fout << d[i] << " ";

    return 0;
}