Cod sursa(job #2990700)

Utilizator InsanekktVlad Matei Insanekkt Data 8 martie 2023 12:50:15
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
#include <tuple>
#include <climits>

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

const int NMAX = 200005, INF=INT_MAX;
vector <pair<int, int> > G[NMAX];
priority_queue<pair<int, int> > pq;
bitset <NMAX> viz;
int n, m, s, tata[NMAX], d[NMAX];

void citire()
{
    fin>>n>>m;
    for(int i=0; i<m; ++i)
    {
        int x, y, c;
        fin>>x>>y>>c;
        G[x].push_back(make_pair(y, c));
    }
    for(int i=1; i<=n; ++i)
        d[i]=INF;
    d[1]=0;
}

void apm()
{
    pq.push(make_pair(0, 1));
    while(!pq.empty())
    {
        int nod=pq.top().second;
        int cost=-pq.top().first;
        pq.pop();
        if(viz[nod])
            continue;
        viz[nod]=1;
        for(auto nbr: G[nod])
            if(!viz[nbr.first])
            {
                if(cost+nbr.second<d[nbr.first]) d[nbr.first]=cost+nbr.second;
                pq.push(make_pair(-cost-nbr.second, nbr.first));
            }
    }
    for(int i=2; i<=n; ++i)
        if(d[i]!=INF)
            fout<<d[i]<<" ";
        else fout<<0<<" ";
}

int main()
{
    citire();
    apm();
}