蒋爹的DSU(非常神奇)
struct DSU {
std::vector<int> f, siz;
DSU(int n) : f(n), siz(n, 1) { std::iota(f.begin(), f.end(), 0); }//f存的是祖宗节点,siz存的是长度
int leader(int x) {//找祖宗
while (x != f[x]) x = f[x] = f[f[x]];
return x;
}
bool same(int x, int y) { return leader(x) == leader(y); }//判断祖宗是否一致
bool merge(int x, int y) {将y节点合并到x节点
x = leader(x);
y = leader(y);
if (x == y) return false;
siz[x] += siz[y];
f[y] = x;
return true;
}
int size(int x) { return siz[leader(x)]; }//求x节点的长度
};