Cod sursa(job #2153434)

Utilizator george.ursachiUrsachi George george.ursachi Data 6 martie 2018 10:43:26
Problema Algoritmul lui Dijkstra Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 2.17 kb
#include <fstream>
#include <vector>
#include <set>

using namespace std;

ifstream f("dijkstra.in");
ofstream g("dijkstra.out");

int N,M,dist[50005];
const int INF = 0x3f3f3f3f;
vector <pair< int, int> > v[50005];
set <pair<int, int> > s;
vector<pair<int, int> > ::iterator it;

void citire()
{
    //citim n,m si apoi m linii
    f>>N>>M;
    int x,y,cost;
    for(int i=1;i<=M;i++)
    {
        //adaugam intr-un vector de perechi care va tine minte vecinul si costul
        f>>x>>y>>cost;
        v[x].push_back(make_pair(y,cost));
    }
}

void Djikstra(int nodInitial)
{
    int nodCurent,vecin,cost;
    //toate dinstantele devin infinit
    for(int i=1;i<=N;i++)
        dist[i]=INF;
    //distanta de la nodul initial la nodul initial este 0
    dist[nodInitial]=0;
    //adaugam nodul initial in setul cate va tine minte cel mai scurt drum
    s.insert(make_pair(nodInitial,0));
    //cat timp mai sunt elemente in set
    while(!s.empty())
    {
        //nod curent ia valoarea primului nod din set(ordine crescatoare)
        nodCurent=s.begin()->first;
        //stergem primul nod din set
        s.erase(s.begin());
        //pentru fiecare vecin al nodului curent
        for(it=v[nodCurent].begin();it!=v[nodCurent].end();it++)
        {
            vecin=it->first;
            cost=it->second;
            //daca noua distanta este mai buna decat cea veche
            if(dist[vecin]>dist[nodCurent]+cost)
            {
                //daca s-a mai ajuns in nodul acesta se sterge din set
                if(dist[vecin]!=INF)
                    s.erase( s.find( make_pair(vecin,dist[vecin]) ) );
                //distanta se actualizeaza iar apoi se adauga in set nodul;
                dist[vecin]=dist[nodCurent]+cost;
                s.insert( make_pair(vecin,dist[vecin]) );
            }
        }
    }
}

void afisare()
{
    //afisarea distantei de la nodul cerut la toate celalalte noduri
    for(int i=2;i<=N;i++)
    {
        if(dist[i]==INF)g<<0<<' ';
        else g<<dist[i]<<' ';
    }
}

int main()
{
    citire();
    Djikstra(1);
    afisare();

    f.close();
    g.close();
    return 0;
}