aldehyde 0.2.158 → 0.2.160

package/src/aldehyde/utils/dsu.ts

@@ -0,0 +1,27 @@
+const N = 1e5 + 10;
+let f;
+
+export const init = (len: number = N) => {
+  f = Array.from({ length: len })
+    .fill(0)
+    .map((i, idx) => idx);
+};
+
+export const find = (x) => {
+  if (f[x] !== x) return (f[x] = find(f[x]));
+  return f[x];
+};
+
+export const merge = (a: number, b: number) => (f[find(a)] = find(b));
+
+export const statistic = () => {
+  const mp = {};
+
+  for (let i = 0; i < N; i++) {
+    const target = find(i);
+    if (mp[target] === undefined) mp[target] = [];
+    mp[target].push(i);
+  }
+
+  return mp;
+};