#ifndef BREADTH_FIRST_TREE_H_
#define BREADTH_FIRST_TREE_H_

#include "out_tree.h"
#include "algorithm/breadth_first_search.h"
using namespace graph;

namespace tree {

template<typename T>
class BreadthFirstTree : public OutTree<T> {
public:
  BreadthFirstTree() = default;

  BreadthFirstTree(Digraph<T> G, T root);

  BreadthFirstTree(std::map<T, std::set<T>> G)
    : OutTree<T>::OutTree(G) {}
};

template<typename T>
BreadthFirstTree<T>::BreadthFirstTree(Digraph<T> G, T root) {
  auto bfs = algo::BreadthFirstSearch<T>(G).execute(root);
  Graph<T>::adjMatrix = bfs;
}

} // namespace tree

#endif