Cod sursa(job #3186606)

Utilizator and_Turcu Andrei and_ Data 23 decembrie 2023 21:17:09
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.08 kb
#include <bits/stdc++.h>
#define N 50007
using namespace std;

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

int n,m;
vector< pair<int,int> >edges[N];
vector<int> dist(N,-1);
vector<bool> viz(N,0);

void Citire()
{
    fin >> n >> m;
    for(int i=1;i<=m;i++)
    {
        int x,y,c;
        fin >> x >> y >> c;
        edges[x].push_back({c,y});
    }
    fin.close();
}

void Dijkstra()
{
    priority_queue<pair<int,int>> pq;
    pq.push({0,1});
    dist[1]=0;
    while( !pq.empty() )
    {
        int cur=pq.top().second;
        pq.pop();

        if( !viz[cur] )
        {
            viz[cur]=1;
            for( auto w:edges[cur] )
            {
                int nod=w.second;
                int cost=w.first;
                if( dist[nod]==-1 or dist[nod]>dist[cur]+cost )
                {
                    dist[nod]=dist[cur]+cost;
                    pq.push({-dist[nod],nod});
                }
            }
        }

    }
}

int main()
{
    Citire();
    Dijkstra();
    for(int i=2;i<=n;i++)
        fout << dist[i] << " ";
    fout.close();
    return 0;
}