Pagini recente » Cod sursa (job #61550) | Cod sursa (job #2301140) | Cod sursa (job #789909) | Cod sursa (job #1648563) | Cod sursa (job #2816103)
// 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 INFINIT INT_MAX
using namespace std;
ifstream fin("dfs.in");
ofstream fout("dfs.out");
struct muchie {
int destinatie, cost, capacitate;
muchie(int destinatie, int cost, int capacitate) : destinatie(destinatie), cost(cost), capacitate(capacitate) {}
};
class Graf {
private:
int nrNoduri, nrMuchii;
bool eOrientat, arePonderi, areCapacitati;
vector<vector<muchie>> listaDeAdiacenta; // lista de vecini
public:
Graf(const int nrNoduri, const int nrMuchii, const bool eOrientat, const bool arePonderi, const bool areCapacitati);
void citire();
vector<int> numarMinimArce(int &nodPlecare);
int numarComponenteConexe();
void afis();
~Graf();
private:
void BFSrecursiv(queue<int> &coada, vector<int> &vizitat);
void DFSrecursiv(int &nodPlecare, 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: listaDeAdiacenta[nodPlecare])
if (vizitat[i.destinatie] == -1) {
// caut toate nodurile nevizitate care sunt adiacente cu nodul de plecare
vizitat[i.destinatie] = vizitat[nodPlecare] + 1; // il marcam vizitat
coada.push(i.destinatie); // il adaug in coada PUSH
}
coada.pop();
BFSrecursiv(coada, vizitat);
}
}
vector<int> Graf::numarMinimArce(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)
vizitat[i] -= 1;
}
return vizitat;
}
void Graf::DFSrecursiv(int &nodPlecare, vector<int> &vizitat) {
vizitat[nodPlecare] = 1;
for (auto &i: listaDeAdiacenta[nodPlecare])
if (!vizitat[i.destinatie])
DFSrecursiv(i.destinatie, vizitat);
}
int Graf::numarComponenteConexe() {
vector<int> nivel(nrNoduri + 1, 0);
vector<int> vizitat(nrNoduri + 1, 0);
nivel[1] = 0;
int numar = 0;
for (int i = 1; i <= nrNoduri; i++)
if (vizitat[i] == 0) {
numar++;
DFSrecursiv(i, vizitat);
}
return numar;
}
void rezolvaBFS() {
// 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, nrMuchii, true, false, false);
g1.citire();
vector<int> distante = g1.numarMinimArce(nodPlecare);
for (int i = 1; i < distante.size(); i++)
fout << distante[i] << " ";
}
void rezolvaDFS() {
// Problema DFS (100p)
// Link: https://infoarena.ro/problema/dfs
// Sursa: https://infoarena.ro/job_detail/2797669?action=view-source
int nrNoduri, nrMuchii;
fin >> nrNoduri >> nrMuchii;
Graf g1(nrNoduri, nrMuchii, false, false, false);
g1.citire();
int numarComponenteConexe = g1.numarComponenteConexe();
fout << numarComponenteConexe;
}
int main() {
int optiune = 2;
switch (optiune) {
case 1:
rezolvaBFS();
case 2:
rezolvaDFS();
}
return 0;
}
void Graf::afis() {
for (int i = 1; i <= nrNoduri; i++) {
fout << "Nodul " << i << " este adiacent cu: ";
for (auto &j: listaDeAdiacenta[i])
fout << j.destinatie << " cu costul " << j.cost << " cu capacitatea " << j.capacitate << "\n";
fout << "break\n";
}
}
void Graf::citire() {
int sursa, destinatie, cost, capacitate;
for (int i = 1; i <= nrMuchii; i++) {
fin >> sursa >> destinatie;
if (arePonderi)
fin >> cost;
else
cost = 0;
if (areCapacitati)
fin >> capacitate;
else
capacitate = 0;
listaDeAdiacenta[sursa].push_back(muchie(destinatie, cost, capacitate));
}
}
Graf::Graf(const int nrNoduri, const int nrMuchii, const bool eOrientat, const bool arePonderi,
const bool areCapacitati) {
this->nrNoduri = nrNoduri;
this->nrMuchii = nrMuchii;
this->eOrientat = eOrientat;
this->arePonderi = arePonderi;
this->areCapacitati = areCapacitati;
listaDeAdiacenta.resize(nrNoduri + 1);
}
Graf::~Graf() {
listaDeAdiacenta.clear();
}