UnionFind.h

#pragma once
#include <cstddef>
#include <cstdint>
#include <vector>
 
namespace enzo::util {
 
class UnionFind
{
  public:
    explicit UnionFind(std::size_t elementCount);
 
    std::size_t find(std::size_t x);
 
    bool unite(std::size_t a, std::size_t b);
 
    bool sameSet(std::size_t a, std::size_t b) { return find(a) == find(b); }
 
    std::size_t componentCount() const { return componentCount_; }
 
    std::size_t size() const { return parent_.size(); }
 
  private:
    std::vector<std::size_t> parent_;
    std::vector<std::uint8_t> rank_;
    std::size_t componentCount_;
};
 
} // namespace enzo::util