Cod sursa(job #2375712)

Utilizator PrEaDiVviNAlexandru Hutu PrEaDiVviN Data 8 martie 2019 11:40:21
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.23 kb
#include <bits/stdc++.h>

using namespace std;

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

#define nmax 50002
///reprezinta o variabila foarte mare, cea mai mare ce o poate memora int ul
const int inf=(1<<31)-1;
int d[nmax];
bool viz[nmax];

struct cmp
{
    ///se ocupa de punerea in coada de prioritate in ordine dupa distante
    bool operator()(int x,int y)
    {
        return d[x]>d[y];
    };
};

vector < pair <int, int> > v[250001];
priority_queue<int, vector<int> , cmp>q;

int n,m;

void dijkstra(int sursa)
{
    ///toare distante sunt initial egale cu infinit
    for(int i=1;i<=n;++i)
        d[i]=inf;
    ///reinitializam distanta de plecare cu 0, intrucat distanta de la un nod la el insusi este 0
    d[sursa]=0;viz[sursa]=1;
    ///punem in coada de prioritati nodul si incepem dijkstra
    q.push(sursa);
    while(!q.empty())
    {
        ///luam nodul cu lungimea cea mai mica si il eliminam din coada
        int nod=q.top();
        viz[nod]=0;
        q.pop();
        ///la nodul actual cautam toate legaturile si verificam daca distanta pana la vecin> decat distanta pana la nod + costul muchiei
        ///daca se intampla o schimbam
        ///Practic el actualizeaza distanta de la un nod pana la celelalte noduri verificand daca legatura ce o gaseste acum este mai mica decat cele gasite anterior
        for(size_t i=0;i<v[nod].size();++i)
        {
            int vecin=v[nod][i].first;
            int cost=v[nod][i].second;
            if(d[vecin]>d[nod]+cost)
            {
                d[vecin]=d[nod]+cost;
                if(!viz[vecin])
                {
                    q.push(vecin);
                    viz[vecin]=1;
                }
            }
        }
    }
}

int main()
{
    ///Algoritmul lui Dijkstra este unul de tip greedy
    ///citire
    f>>n>>m;
    for(int i=1;i<=m;++i)
    {
        int x,y,c;
        f>>x>>y>>c;
        ///pun la nodul x legatura y de cost c
        ///astfel oarecum se creeaza o matrice cu diferite lungimi
        v[x].push_back(make_pair(y,c));
    }
    dijkstra(1);
    for(int i=2;i<=n;i++)
        if(d[i]!=inf)g<<d[i]<<" ";
        else g<<"0 ";
    f.close();g.close();
    return 0;
}