Logo

Programming-Idioms

This language bar is your friend. Select your favorite languages!

Idiom #130 Depth-first traversal in a graph

Call th function f on every vertex accessible from the vertex v, in depth-first prefix order

#include <functional>
#include <stack>
#include <unordered_set>
void dfs(Node& root, std::function<void(Node*)> f) {
  std::stack<Node*> queue;
  queue.push(&root);
  std::unordered_set<const Node*> visited;
  while (!queue.empty()) {
    Node* const node = queue.top();
    queue.pop();
    f(node);
    visited.insert(node);
    for (Node* const neighbor : node->neighbors) {
      if (visited.find(neighbor) == visited.end()) {
        queue.push(neighbor);
      }
    }
  }
}

New implementation...