lisyarus/psemek
1
0
Fork 1
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve util::dfs::cycle_iterator

Browse source
This commit is contained in:
Nikita Lisitsa 2023-04-15 13:16:09 +03:00
parent 3bfe867cc0
commit df5e188236
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
libs/util/include/psemek/util/dfs.hpp
View file

@ -20,20 +20,39 @@ namespace psemek::util
struct cycle_iterator
{
using iterator_category = std::forward_iterator_tag;
using value_type = Index;
using reference_type = Index;
using pointer = Index *;
using difference_type = int;
dfs * parent;
Index i;
Index operator *() const { return i; }
cycle_iterator & operator ++()
{
i = parent->parent_[i];
return *this;
}
cycle_iterator operator ++(int)
{
auto copy = *this;
i = parent->parent_[i];
return copy;
}
friend bool operator == (cycle_iterator const & i1, cycle_iterator const & i2)
{
return i1.i == i2.i;
}
friend bool operator != (cycle_iterator const & i1, cycle_iterator const & i2)
{
return i1.i != i2.i;
}
};
template <typename CycleCallback>