Cod sursa(job #2802983)

Utilizator ptr22222Petru Popescu ptr22222 Data 19 noiembrie 2021 10:30:43
Problema Arbore partial de cost minim Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>

using namespace std;

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

const int nmax = 100001;

class Graf
{
private:
    bool orientat;
    int nrNoduri;
    vector <int> listaAdiacenta[nmax];
public:
    Graf(int nrNoduri = 0, bool orientat = false);
    int get_nrNoduri();
    void citireMuchii(int nrMuchii);
    vector<int> BFS(int start);
    void distanteMinime(int start);
    void DFS(int nodCurent, bool *vizitatDFS);
    int nrComponenteConexe();
};

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

int Graf::get_nrNoduri()
{
    return this->nrNoduri;
}

void Graf::citireMuchii(int nrMuchii)
{
    int nod1, nod2;
    for(int i = 0; i < nrMuchii; i++)
    {
        in >> nod1 >> nod2;
        listaAdiacenta[nod1].push_back(nod2);
        if(!orientat)
        {
            listaAdiacenta[nod2].push_back(nod1);
        }
    }
}

vector<int> Graf::BFS(int start)
{
    queue <int> coadaBFS;
    int vizitatBFS[nmax] = {false};
    vector <int> distantaBFS(nmax);
    fill(distantaBFS.begin(), distantaBFS.end(), 0);
    coadaBFS.push(start);
    vizitatBFS[start] = true;
    distantaBFS[start] = 1;
    while(!coadaBFS.empty())
    {
        int nodCurent = coadaBFS.front();
        coadaBFS.pop();
        for(auto vecin:listaAdiacenta[nodCurent])
        {
            if(!vizitatBFS[vecin])
            {
                vizitatBFS[vecin] = true;
                distantaBFS[vecin] = distantaBFS[nodCurent] + 1;
                coadaBFS.push(vecin);
            }
        }
    }
    return distantaBFS;
}

void Graf::distanteMinime(int start)
{
    vector <int> res;
    res = BFS(start);
    for(int i = 1; i <= nrNoduri; i++)
    {
        out << res[i] - 1 << " ";
    }
    out <<"\n";
}

int main() {
    int n, m, start;
    in >> n >> m >> start;
    Graf g(n,true);
    g.citireMuchii(m);
    g.distanteMinime(start);
    return 0;
}