Cod sursa(job #3186591)

Utilizator and_Turcu Andrei and_ Data 23 decembrie 2023 19:57:21
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.91 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);

void Citire()
{
    fin >> n >> m;
    for(int i=1;i<=n;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});
    while( !pq.empty() )
    {
        int x,cost;
        x=pq.top().second;
        cost=-pq.top().first;
        pq.pop();
        if( dist[x]!=-1 )continue;
        dist[ x ]=cost;
        for( auto w:edges[x] )
        {
            int y=w.second;
            int c=w.first+cost;
            if( dist[y]==-1 )pq.push( {-c,y} );
        }
    }
}

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