Cod sursa(job #2803136)

Utilizator redikusTiganus Alexandru redikus Data 19 noiembrie 2021 15:55:49
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 3.31 kb
#include <iostream>
#include <fstream>
#include <bits/stdc++.h>

using namespace std;

class graf {

protected:
    int noduri, muchii;
    vector < vector < int >> adiacenta;

public:
    // Constructor cu matrice de adiacenta data
    graf(vector < vector < int >> listaAdiacenta, int noduri, int muchii): adiacenta(listaAdiacenta), noduri(noduri), muchii(muchii) {};

    // Constructor fara matrice de adiacenta, se cu initalizeaza una goala
    graf(int noduri, int muchii): adiacenta(noduri + 1), noduri(noduri), muchii(muchii) {};

    vector < int > bfs(int);
    int dfs();

};

class graf_neorientat : public graf{
private:
    void dfsbiconex(int, vector < int >&, vector < int >&, stack < pair < int, int >>&, int&, vector < string >&);
    void dfsCriticalConnections(int tata, vector < int >&, vector < int >&, int&, vector < vector < int >>&, vector < int >&);

public:
    graf_neorientat(vector < vector < int >> listaAdiacenta, int noduri, int muchii): graf(listaAdiacenta, noduri, muchii) {};
    graf_neorientat(int noduri, int muchii): graf(noduri, muchii) {};

    friend istream& operator>>(istream&, graf_neorientat&);

    vector < string > biconex();

    vector < vector < int >> criticalConnections();
};

class graf_neorientat_ponderat : public graf_neorientat {
    // Matrice de adianceta care contine si costurile
    vector< vector< pair< int, int >>> adiacentap;
public:
    graf_neorientat_ponderat(vector < vector < int >> listaAdiacenta, int noduri, int muchii): graf_neorientat(listaAdiacenta, noduri, muchii) {};
    graf_neorientat_ponderat(int noduri, int muchii): adiacentap(noduri+1), graf_neorientat(noduri, muchii) {};

    friend istream& operator>>(istream&, graf_neorientat_ponderat&);

    vector<int> dijkstra();

};

istream& operator>>(istream& in, graf_neorientat_ponderat& g) {
    for (int i = 0; i < g.muchii; i++) {
        int aux1, aux2, aux3;
        in >> aux1 >> aux2 >> aux3;
        g.adiacentap[aux1].push_back(make_pair(aux2, aux3));
        g.adiacentap[aux2].push_back(make_pair(aux1, aux3));
    }
    return in;
}

vector<int> graf_neorientat_ponderat :: dijkstra() {


    priority_queue<vector<int>> p;
    vector<int> m(noduri+1, INT_MAX);
    vector<bool> vis(noduri+1, false);

    m[1] = 0;
    p.push({0, 1});
    while(!p.empty()){
        int u = p.top()[1];
        p.pop();
        if(!vis[u]){
            vis[u] = true;
            for(auto i : adiacentap[u]){
                int v = i.first, cost = i.second;
                if(!vis[v] && m[v] > m[u] + cost){
                    //cout<<i.first<<" "<<m[v]<<endl;
                    m[v] = m[u] + cost;
                    p.push({-(m[v]), v});
                }
            }
        }
    }
    return m;

}

void dijkstraDriver() {
    // https://infoarena.ro/problema/dijkstra
    // Citire
    ifstream in ("dijkstra.in");
    ofstream out("dijkstra.out");

    int n, m;
    in >> n >> m;

    graf_neorientat_ponderat x(n, m);
    in>>x;
    vector<int> v = x.dijkstra();
    for(int i=2; i<=n; i++){
        if(v[i] == INT_MAX){
            out<<"0 ";
        }
        else{
            out<<v[i]<<" ";
        }
    }

}

int main() {
    // Se apeleaza functia corespunzatoare problemei
    dijkstraDriver();
    return 0;
}