Cod sursa(job #2730826)

Utilizator xXoctavianXxStanescu Matei Octavian xXoctavianXx Data 26 martie 2021 22:00:13
Problema Algoritmul lui Dijkstra Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.02 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 vizitat[nmax+2];

void dijkstra(int start)
{
    pq.push({0,start});
    while(!pq.empty())
    {
        ll cost=pq.top().first;
        int i=pq.top().second;
        vizitat[i]=cost;
        pq.pop();
        for(int it=0; it<vecini[i].size(); it++)
        {
            int pi=vecini[i][it].first;
            if(vizitat[pi]==0 || vizitat[pi]>cost+vecini[i][it].second)
            {
                pq.push({cost+vecini[i][it].second,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<<vizitat[i]<<" ";
    }
    return 0;
}