Cod sursa(job #2038263)

Utilizator vancea.catalincatalin vancea.catalin Data 13 octombrie 2017 15:57:09
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.18 kb
#include<iostream>
#include<fstream>
#include<vector>
#include<set>
#define DN 50010
using namespace std;
fstream fin("dijkstra.in",ios::in),fout("dijkstra.out",ios::out);
struct strv
{
public:
    int b,c;
};
struct strh
{
public:
    int d,a;
    bool operator < (const strh &y) const
    {
        return d<y.d;
    }
    void operator = (multiset<strh> :: iterator it)
    {
        //std::cout<<"adresa: "<<it<<"\n";
        d=(*it).d;a=(*it).a;
        //return {(*it).d,(*it).a};
    }
};
vector<strv> v[DN];
multiset<strh> heap,aux;
int n,m,a,b,c,dist[DN],ap[DN];
int main()
{
    int n,i,x,d,nod;

    strh it;
    fin>>n>>m;
    for(i=1;i<=m;i++)
    {
        fin>>a>>b>>c;
        v[a].push_back({b,c});
    }

    heap.insert({0,1});
    it=heap.begin();

    while(heap.empty()==0)
    {
        it=heap.begin();
        d=it.d; nod=it.a;
        heap.erase(heap.begin());
        if(ap[nod]==1) continue;
        ap[nod]=1;
        for(auto j:v[nod])
        {
            if((dist[j.b]>d+j.c) || (dist[j.b]==0 && ap[j.b]==0))
            {
                dist[j.b]=d+j.c;
                heap.insert({dist[j.b],j.b});
            }
        }
    }
    for(i=2;i<=n;i++) fout<<dist[i]<<" ";
}