并查集的各种模板（持续更新）

平庸并查集的初始化

void init(ll n)
{
    for(ll i=1;i<=n;i++)
    {
        fa[i]=i;
    }
}

平庸并查集的查找

ll find(ll x)
{
    if(fa[x]==x)
        return x;
    else find(fa[x]);
}

平庸并查集的合并

void merge(int i,int j)
{
    fa[find(i)]=find(j);
}

查找的路径压缩优化

ll find(ll x)
{
    if(fa[x]==x)
        return x;
    else{
        fa[x]=find(fa[x]);  //父节点设为根节点,减少后续查找时间
        return fa[x];   //返回父节点
    }
}

查找的路径压缩优化（代码简写版）

ll find(ll x)
{
    return x==fa[x]?x:(fa[x]=find(fa[x]));
}

合并的按秩合并优化，这是需要初始化rnk数组

void init(ll n)
{
    for(ll i=1;i<=n;i++)
    {
        fa[i]=i;
        rnk[i]=1;
    }
}
void merge(ll i,ll j)
{
    ll x=find(i);
    ll y=find(j);
    if(rnk[x]<=rnk[y]){
        fa[x]=y;
    }
    else{
        fa[y]=x;
    }
    if(x!=y && rnk[x]==rnk[y])
    {
        rnk[y]++;
    }
}

用并查集统计集合的个数

int count_sets(int n)
{
	int cnt = 0;
	for (int i = 1; i <= n; i++)
		if (find(i) == i)
			cnt++;
 
	return cnt;
}

用并查集统计一个集合内节点的个数，增加num数组记录一个集合内元素的个数

void init(ll n)
{
    for(ll i=1;i<=n;i++)
    {
        fa[i]=i;
        rnk[i]=1;
        num[i]=1;
    }
}
void merge_num(ll i,ll j)
{
    ll x=find(i);
    ll y=find(j);
    if(rnk[x]<=rnk[y]){
        fa[x]=y;
        if(x!=y)
            num[y]+=num[x];
    }
    else{
        fa[y]=x;
        if(x!=y)
            num[x]+=num[y];
    }
    if(x!=y && rnk[x]==rnk[y])
    {
        rnk[y]++;
    }
}

当数据量过大时用map优化并查集

unordered_map<ll,ll> Fa,m;
ll find_map(ll x)
{
    return x==Fa[x]?x:(Fa[x]=find_map(Fa[x]));
}
void merge(ll i,ll j)
{
    ll x=find_map(i);
    ll y=find_map(j);
    if(rnk[x]<=rnk[y]){
        Fa[x]=y;
    }
    else{
        Fa[y]=x;
    }
    if(x!=y && rnk[x]==rnk[y])
    {
        rnk[y]++;
    }
}