From 83f92ae9cbf9b2bf081ceabe026c13287308a116 Mon Sep 17 00:00:00 2001
From: lisyarus <lisyarus@gmail.com>
Date: Mon, 18 Mar 2024 15:06:07 +0300
Subject: [PATCH] Add util::dsu

---
 libs/util/include/psemek/util/dsu.hpp | 54 +++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 libs/util/include/psemek/util/dsu.hpp

diff --git a/libs/util/include/psemek/util/dsu.hpp b/libs/util/include/psemek/util/dsu.hpp
new file mode 100644
index 00000000..63765875
--- /dev/null
+++ b/libs/util/include/psemek/util/dsu.hpp
@@ -0,0 +1,54 @@
+#pragma once
+
+#include <vector>
+#include <cstdint>
+
+namespace psemek::util
+{
+
+	template <typename Index = std::uint32_t>
+	struct dsu
+	{
+		dsu() = default;
+		dsu(Index count);
+
+		void combine(Index x, Index y);
+		Index find(Index x);
+
+	private:
+		std::vector<Index> parent_;
+		std::vector<Index> size_;
+	};
+
+	template <typename Index>
+	dsu<Index>::dsu(Index count)
+		: parent_(count)
+		, size_(count, 1)
+	{
+		for (Index i = 0; i < count; ++i)
+			parent_[i] = i;
+	}
+
+	template <typename Index>
+	void dsu<Index>::combine(Index x, Index y)
+	{
+		x = find(x);
+		y = find(y);
+		if (x != y)
+		{
+			if (size_[x] < size_[y])
+				std::swap(x, y);
+			parent_[y] = x;
+			size_[x] += size_[y];
+		}
+	}
+
+	template <typename Index>
+	Index dsu<Index>::find(Index x)
+	{
+		if (parent_[x] == x)
+			return x;
+		return parent_[x] = find(parent_[x]);
+	}
+
+}