I was trying to implement BFS using Vertex classes and BFS class, since I wanted to learn implementation of classes along with algorithms.
vertex.h
#ifndef vertex_H_
#define vertex_H_
#include<vector>
class Vertex{
private:
int _data;
bool _visited;
std::vector<Vertex> _neighbours;
public:
Vertex(int data);
Vertex();
void setData(int data);
int getData();
bool isVisited();
void setVisited(bool visited);
std::vector<Vertex> getNeighbours();
void setNeighbours(std::vector<Vertex>& neighbours);
void addNeighbourVertex(Vertex& vertex);
~Vertex();
};
#endif
vertex.cpp
#include "vertex.h"
Vertex::Vertex(int data):_data(data){}
Vertex::Vertex(){}
Vertex::~Vertex(){}
void Vertex::setData(int data){
_data = data;
}
int Vertex::getData(){
return _data;
}
void Vertex::setVisited(bool visited){
_visited=visited;
}
bool Vertex::isVisited(){
return _visited;
}
std::vector<Vertex> Vertex::getNeighbours(){
return _neighbours;
}
void Vertex::setNeighbours(std::vector<Vertex>& neighbours){
_neighbours = neighbours;
}
void Vertex::addNeighbourVertex(Vertex& vertex){
_neighbours.push_back(vertex);
}
BFS.h
#ifndef BFS_H_
#define BFS_H_
#include "vertex.h"
class BFS{
public:
void bfs(Vertex& root);
};
#endif
BFS.cpp
#include<iostream>
#include<vector>
#include"vertex.h"
#include<deque>
#include"BFS.h"
void BFS::bfs(Vertex& root)
{
std::deque<Vertex> Q;
root.setVisited(true);
Q.push_back(root);
while(!Q.empty())
{
Vertex select_node = Q.front();
Q.pop_front();
std::cout<<select_node.getData()<<" "<<std::endl;
for(Vertex node : select_node.getNeighbours())
{
if (!node.isVisited())
{
node.setVisited(true);
Q.push_back(node);
}
}
}
}
application.cpp
#include<iostream>
#include "BFS.h"
#include "vertex.h"
int main(){
BFS bfs;
Vertex v1(1);
Vertex v2(2);
Vertex v3(3);
Vertex v4(4);
Vertex v5(5);
v1.addNeighbourVertex(v2);
v1.addNeighbourVertex(v4);
v4.addNeighbourVertex(v5);
v2.addNeighbourVertex(v3);
bfs.bfs(v1);
return 0;
}
- getNeighbours function from vertex class gets the vector of neighbouring vertices of particular vertex.
Problem I am facing is that for graph in application.cpp that I have defined.
vertex 1-> vertices{2,4}
vertex 2-> vertex{3}
vertex 3-> vertex{2}
vertex 4-> vertex{5}
vertex 5-> vertex{4}
expected output is: 1 2 4 3 5
My output : 1 2 4
I would really like some assistance.. there is some thing i am unable to see here, and hence not getting the proper output.