minor change for dfs/bfs

minor change when element is not in graph when dfs/bfs-ing.
spirosmaggioros · Jan 20, 2024 · 02ca008 · 02ca008
1 parent 9a87090
commit 02ca008
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/classes/graph/graph.h b/src/classes/graph/graph.h
@@ -89,7 +89,7 @@ template<typename T> size_t graph<T>::size(){return __elements.size();}
 
 template <typename T> std::vector<T> graph<T>::dfs(T start) {
   std::vector<T> path;
-  if(this -> empty()){
+  if(this -> empty() || __elements.find(start) == __elements.end()){
     return path;
   }
   std::stack<T> s;
@@ -112,7 +112,7 @@ template <typename T> std::vector<T> graph<T>::dfs(T start) {
 
 template <typename T> std::vector<T> graph<T>::bfs(T start) {
   std::vector<T> path;
-  if(this -> empty()){
+  if(this -> empty() || __elements.find(start) == __elements.end()){
     return path;
   }
   std::queue<T> q;
@@ -400,7 +400,7 @@ template <typename T> int64_t weighted_graph<T>::shortest_path(T start, T end) {
 template <typename T> std::vector<T> weighted_graph<T>::dfs(T start) {
 
   std::vector<T> path;
-  if(this -> empty()){
+  if(this -> empty() || __elements.find(start) == __elements.end()){
     return path;
   }
   std::queue<T> q;
@@ -426,7 +426,7 @@ template <typename T> std::vector<T> weighted_graph<T>::dfs(T start) {
 
 template <typename T> std::vector<T> weighted_graph<T>::bfs(T start) {
   std::vector<T> path;
-  if(this -> empty()){
+  if(this -> empty() || __elements.find(start) == __elements.end()){
     return path;
   }
   std::queue<T> q;