Pagini recente » Cod sursa (job #2398581) | Cod sursa (job #2481504) | Cod sursa (job #2263950) | Cod sursa (job #143158) | Cod sursa (job #2806600)
// Popescu Paullo Robertto Karloss Grupa 2311
// Linkurile pentru fiecare problema se gasesc in main
#include <fstream>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <algorithm>
#include <climits>
#define maxi 100001
using namespace std;
ifstream fin("bfs.in");
ofstream fout("bfs.out");
class Graf {
private:
int nrNoduri;
int sursa, destinatie; // extremitate muchie stanga respectiv dreapta
bool eOrientat, arePonderi;
vector<int> *adiacenta; // lista de vecini
vector<vector<pair<int, int>>> adiacentaCost;
// vector<int> *adiacenta2; // lista de vecini transpusa (i.e. in loc de x si y folosim y si x ca la grafuri neorientate)
public:
Graf(const int nrNoduri, const bool eOrientat, const bool arePonderi);
void citire(const int nrMuchii);
void rezolvaBFS(int &nodPlecare);
private:
void adaugaMuchie(const int sursa, const int destinatie);
void adaugaMuchiePonderat(const int sursa, const int destinatie, const int cost);
void BFSrecursiv(queue<int> &coada, vector<int> &vizitat);
};
void Graf::BFSrecursiv(queue<int> &coada, vector<int> &vizitat) {
if (!coada.empty()) // daca mai sunt elemente in coada / nu am verificat pt toate nodurile
{
int nodPlecare = coada.front(); // retin nodul de unde plec
for (auto &i: adiacenta[nodPlecare])
if (vizitat[i] == -1) {
// caut toate nodurile nevizitate care sunt adiacente cu nodul de plecare
vizitat[i] = vizitat[nodPlecare] + 1; // il marcam vizitat
coada.push(i); // il adaug in coada PUSH
}
coada.pop();
BFSrecursiv(coada, vizitat);
}
}
void Graf::rezolvaBFS(int &nodPlecare) {
vector<int> vizitat(nrNoduri + 1, -1);
queue<int> coada;
coada.push(nodPlecare);
vizitat[coada.back()] = 1;
BFSrecursiv(coada, vizitat);
for (int i = 1; i <= nrNoduri; i++) {
if (vizitat[i] == -1)
fout << -1 << " ";
else
fout << vizitat[i] - 1 << " ";
}
}
int main() {
// Tema 1:
// Problema BFS (100p)
// Link: https://infoarena.ro/problema/bfs
// Sursa: https://infoarena.ro/job_detail/2797664?action=view-source
int nrNoduri, nrMuchii, nodPlecare;
fin >> nrNoduri >> nrMuchii >> nodPlecare;
Graf g1(nrNoduri, true, false);
g1.citire(nrMuchii);
g1.rezolvaBFS(nodPlecare);
// g1.BFS();
// g1.afisareCoadaBFS();
fin.close();
fout.close();
return 0;
}
Graf::Graf(const int nrNoduri, const bool eOrientat, const bool arePonderi) {
this->nrNoduri = nrNoduri;
this->eOrientat = eOrientat;
this->arePonderi = arePonderi;
}
void Graf::adaugaMuchie(const int sursa, const int destinatie) {
adiacenta[sursa].push_back(destinatie);
}
void Graf::adaugaMuchiePonderat(const int sursa, const int destinatie, const int cost) {
adiacentaCost[sursa].push_back(make_pair(destinatie, cost));
}
void Graf::citire(const int nrMuchii) {
if (!arePonderi) {
adiacenta = new vector<int>[nrNoduri + 1];
if (eOrientat)
for (int i = 1; i <= nrMuchii; i++) {
fin >> sursa >> destinatie;
adaugaMuchie(sursa, destinatie);
}
else
for (int i = 1; i <= nrMuchii; i++) {
fin >> sursa >> destinatie;
adaugaMuchie(sursa, destinatie);
adaugaMuchie(destinatie, sursa);
}
} else {
int cost;
adiacentaCost.resize(nrNoduri + 1, vector<pair<int, int>>(1, {-1, -1}));
if (eOrientat)
for (int i = 1; i <= nrMuchii; i++) {
fin >> sursa >> destinatie >> cost;
adaugaMuchiePonderat(sursa, destinatie, cost);
}
else
for (int i = 1; i <= nrMuchii; i++) {
fin >> sursa >> destinatie >> cost;
adaugaMuchiePonderat(sursa, destinatie, cost);
adaugaMuchiePonderat(destinatie, sursa, cost);
}
}
}