Cod sursa(job #2189928)

Utilizator BogdanAlexandruBogdan-Alexandru Dig BogdanAlexandru Data 29 martie 2018 13:26:39
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.09 kb
#include <stdio.h>
#include <fstream>
#include <queue>
#include <vector>
#include <climits>
#define mp make_pair
using namespace std;
vector <pair <int,int> >graf[50015];
FILE *f,*g;
int drum[50015];
bool viz[50015];
struct Nod
{
    int v,c;
};
struct compara///functia asta e un criteriu de plasare a elementelor in coada
{
    bool operator()(Nod x,Nod y)///deoarece avem nevoie de DRUMUL MINIM
    {
        return x.c>y.c;
    }
};
priority_queue<Nod, vector<Nod>,compara> coada;

void Dijkstra(int start,int n)
{
    coada.push({1,0});///adaug in coada pozitia de incepere
    int nod,cost,vecin,drm;
    viz[start]=false;
    for(int i=1;i<=n;i++)///formez vectorul de drum minim cu elemente maxime
        drum[i]=INT_MAX;
    drum[start]=0;/// marchez pozitia de incepere ca fiind 0
    while(!coada.empty())/// cat timp am elemente in coada
    {
       nod=coada.top().v;///scot elementul din varful cozii
       drm=coada.top().c;///
       coada.pop();/// elimin elementul din varful cozii
       if(viz[nod]==false)
       {
          for(size_t i=0;i<graf[nod].size();i++)///parcurg vecinii nodului extras
          {
             vecin=graf[nod][i].first;///imi salvez vecinul
             cost=graf[nod][i].second;/// imi salvez costul muchiei
             if(drm+cost<drum[vecin])///daca am venit in nodul vecin cu un drum mai bun
             {
                 drum[vecin]=drm+cost;///actualizez valoarea din nod
                 coada.push({vecin,drum[vecin]});/// si imi pun in coada nodul
             }
          }
       }
       viz[nod]=true;
    }
}
void Afiseaza(int n)
{
    int i;
    for(i=2;i<=n;i++)
        if(drum[i]!=INT_MAX)
            fprintf(g,"%d ",drum[i]);
        else
            fprintf(g,"0 ");
}
int main()
{
    int i,j,n,m,x,y,c;
    f=fopen("dijkstra.in","r");
    g=fopen("dijkstra.out","w");
    fscanf(f,"%d %d",&n,&m);
    for(i=1;i<=m;i++)
    {
        fscanf(f,"%d %d %d",&x,&y,&c);
        graf[x].push_back(mp(y,c));
    }
    Dijkstra(1,n);
    Afiseaza(n);
    fclose(f);
    fclose(g);
    return 0;
}