Cod sursa(job #2929200)

Utilizator k20ySabaceag Marius k20y Data 25 octombrie 2022 10:32:31
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.28 kb
#include <bits/stdc++.h>
using namespace std;

#define pb push_back
#define mp make_pair
#define to(nod,i) la[nod][i].first
#define cost(nod,i) la[nod][i].second
#define pp pair<int,int>

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

const int N=5e4+5;

vector<pair<int,int> > la[N];
int cost[N],dist[N];
bool vizitat[N];

void dijkstra(int start, int n)
{
    for(int i=1;i<=n;i++)
        if(i!=start) dist[i]=INT_MAX;

    priority_queue<pp,vector<pp>,greater<pp> > pq;

    vizitat[start]=1;
    pq.push(mp(0,start));

    while(!pq.empty())
    {
        int nod=pq.top().second;
        int cost=pq.top().first;
        pq.pop();

        for(int i=0;i<la[nod].size();i++)
            if(!vizitat[to(nod,i)])
        {
            vizitat[to(nod,i)]=1;

            if(cost + cost(nod,i) < dist[to(nod,i)])
            {
                dist[to(nod,i)] = cost + cost(nod,i);
                pq.push(mp(dist[to(nod,i)],to(nod,i) ) );
            }
        }
    }

}

int main()
{
    int n,m; in>>n>>m;

    for(int i=1;i<=m;i++)
    {
        int x,y,c; in>>x>>y>>c;

        la[x].pb(mp(y,c));
    }

    dijkstra(1,n);

    for(int i=2;i<=n;i++)
        if(dist[i] == INT_MAX) out<<0<<' ';
    else out<<dist[i]<<' ';

}