Cod sursa(job #2152034)

Utilizator DologXTurcas David Emil DologX Data 5 martie 2018 10:25:00
Problema Algoritmul lui Dijkstra Scor 90
Compilator cpp Status done
Runda Arhiva educationala Marime 1 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <set>

using namespace std;

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

int n,m,a,b,c,cost[50005],ff,ss;
vector < pair<int,int> > v[50005];
set < pair<int,int> > s;

int main()
{
    fin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        fin>>a>>b>>c;
        v[a].push_back(make_pair(b,c));
    }
    m=n;
    for(int i=1;i<=n;i++) cost[i]=-1;
    s.insert(make_pair(0,1));
    while(!s.empty())
    {
        c=(*s.begin()).first;
        n=(*s.begin()).second;
        s.erase(*s.begin());
        if(cost[n]==-1)
        {
            cost[n]=c;
            for(int i=0;i<v[n].size();i++)
            {
                ff=v[n][i].first;
                ss=v[n][i].second;
                if(cost[ff]==-1) s.insert(make_pair(c+ss,ff));
            }
        }
    }
    for(int i=2;i<=m;i++)
    {
        if(cost[i]==-1) fout<<0<<' ';
            else fout<<cost[i]<<' ';
    }
    return 0;
}