Cod sursa(job #2669434)

Utilizator loraclorac lorac lorac Data 6 noiembrie 2020 22:45:29
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include <bits/stdc++.h>
using namespace std;
ifstream in("dijkstra.in");
ofstream out("dijkstra.out");
int n,m,x,y,c;
const int lim=5e4+5;
const int inf=2e9+7;
vector<pair<int,int> > vec[lim];
set<pair<int,int> > s;
int dist[lim];
void dijkstra()
{
    dist[1]=0;
    for(int i=2;i<=n;++i)
        dist[i]=inf;
    s.insert({0,1});
    while(!s.empty())
    {
        int cost=(*(s.begin())).first;
        int nod=(*(s.begin())).second;
        s.erase(s.begin());
        for(auto x:vec[nod])
        if(dist[x.first]>cost+x.second)
        {
            if(dist[x.first]!=inf)
                s.erase(s.find({dist[x.first],x.first}));
            dist[x.first]=cost+x.second;
            s.insert({dist[x.first],x.first});
        }
    }
    for(int i=2;i<=n;++i)
    if(dist[i]==inf) out<<0<<' ';
    else out<<dist[i]<<' ';
}
int main()
{
    in>>n>>m;
    for(int i=1;i<=m;++i)
    {
        in>>x>>y>>c;
        vec[x].push_back({y,c});
    }
    dijkstra();
    return 0;
}