Cod sursa(job #2802636)

Utilizator ptr22222Petru Popescu ptr22222 Data 18 noiembrie 2021 16:10:48
Problema Algoritmul lui Dijkstra Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.96 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <stack>
#include <algorithm>

using namespace std;

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

const int nmax = 50001;
const int inf = (1<<30);

class Graf
{
private:
    bool orientat;
    int nrNoduri;
    vector <int> listaAdiacenta[nmax];
    vector <pair <int, int>> listaAdiacentaCost[nmax];
public:
    Graf(int nrNoduri = 0, bool orientat = false);
    void adaugaMuchieCost(int, int, int);
    void citireMuchii(int);
    void citireMuchiiCost(int);
    vector<int> BFS(int);
    void DFS(int, bool*);
    int nrComponenteConexe();
    void DFSStiva(int nodcurent, bool *, stack<int> &);
    void sortareTopologica();
    void TarjanCTC(int&, int, int*, int*, bool*, stack <int> &, vector<vector<int>> &);
    void CTC();
    void TarjanBC(int&, int, int, int*, int*, stack <int>&, vector<vector<int>>&);
    void BC();
    void TarjanMC(int&, int, int, int*, int*, vector<pair<int,int>>&);
    void MC();
    vector <int> citireHakimi();
    bool Hakimi(vector <int>);
    vector <int> Dijkstra(int);
};

Graf :: Graf(int nrNoduri, bool orientat) : nrNoduri(nrNoduri), orientat(orientat)
{
    this->nrNoduri = nrNoduri;
    this->orientat = orientat;
}

void Graf::adaugaMuchieCost(int nod1, int nod2, int cost)
{
    listaAdiacentaCost[nod1].push_back(make_pair(nod2, cost));
}

void Graf::citireMuchiiCost(int nrMuchii)
{
    int nod1, nod2, cost;
    for(int i = 0; i < nrMuchii; i++)
    {
        in >> nod1 >> nod2 >> cost;
        adaugaMuchieCost(nod1, nod2, cost);
        if(!orientat)
        {
            adaugaMuchieCost(nod2, nod1, cost);
        }
    }
}

vector <int> Graf::Dijkstra(int nodStart)
{
    vector <int> distanta(nmax);
    fill(distanta.begin(), distanta.end(), inf);
    priority_queue <pair <int, int>, vector <pair <int, int>>, greater<pair<int,int>>> coada;
    bool vizitat[nmax] = {0};
    distanta[nodStart] = 0;
    coada.push(make_pair(0,nodStart));
    while(!coada.empty())
    {
        int nodCurent = coada.top().second;
        coada.pop();
        if(!vizitat[nodCurent])
        {
            vizitat[nodCurent] = true;
            for(auto vecin:listaAdiacentaCost[nodCurent])
            {
                if(!vizitat[vecin.first])
                {
                    if(distanta[nodCurent] + vecin.second < distanta[vecin.first])
                    {
                        distanta[vecin.first] = distanta[nodCurent] + vecin.second;
                        coada.push(make_pair(distanta[vecin.first], vecin.first));
                    }
                }
            }
        }
    }
    return distanta;
}

int main()
{
    int n, m;
    in >> n >> m;
    Graf g(n,true);
    g.citireMuchiiCost(m);
    vector <int> rezultat(nmax);
    rezultat = g.Dijkstra(1);
    for(int i = 2; i <=n; i++)
    {
        if(rezultat[i] == inf)
        {
            out << 0 << " ";
            continue;
        }
        out << rezultat[i] << " ";
    }
    return 0;
}