Cod sursa(job #2730831)

Utilizator xXoctavianXxStanescu Matei Octavian xXoctavianXx Data 26 martie 2021 22:15:28
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.94 kb
#include <bits/stdc++.h>
#include <queue>
#define ll long long

using namespace std;

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

const int nmax=50000;
const int mmax=250000;
int n,m,a,b,c;
priority_queue<pair<ll,int> > pq;
vector<pair<int,int> > vecini[nmax+2];
ll d[nmax+2];

void dijkstra(int start)
{
    pq.push({0,start});
    while(!pq.empty())
    {
        int i=pq.top().second;
        pq.pop();
        for(auto nod: vecini[i])
        {
            int pi=nod.first;
            int cost=nod.second;
            if(d[pi]==0 || d[pi]>d[i]+cost)
            {
                d[pi]=d[i]+cost;
                pq.push({d[pi],pi});
            }
        }
    }
}

int main()
{
    fin>>n>>m;
    for(int i=0; i<n; i++)
    {
        fin>>a>>b>>c;
        vecini[a].push_back({b,c});
    }
    dijkstra(1);
    for(int i=2; i<=n; i++)
    {
        fout<<d[i]<<" ";
    }
    return 0;
}