并查集学习

并查集学习

概念：
将编号分别为1~n的n个对象划分为不相交集合，在每个集合中，选择其中某个元素代表所在集合。在这个集合中，并查集的操作有初始化、合并、查找。

题目
“How Many Tables"
有n个人一起吃饭，有些人互相认识。认识的人想坐在一起，不想跟陌生人坐。例如A认识B，B认识C，那么A、B、C会坐在一张桌子上。
给出认识的人，问需要多少张桌子。

题解：
一张桌子是一个集，合并朋友关系，然后统计集的数量即可。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1050;
int s[maxn];
void init_set()
{
    for(int i=1;i<=maxn;i++)
    {
        s[i]=i;
    }
}
int find_set(int x)
{
    return x==s[x]?x:find_set(s[x]);
}
void union_set(int x,int y)
{
    x=find_set(x);
    y=find_set(y);
    if(x!=y)s[x]=s[y];
}
int main()
{
    int t,n,m,x,y;
    cin>>t;
    while(t--)
    {
        cin>>n>>m;
        init_set();
        for(int i=1;i<=m;i++)
        {
            cin>>x>>y;
            union_set(x,y);

        }
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            if(s[i]==i)
                ans++;

        }
        cout<<ans<<endl;
    }
    return 0;
}

优化：
合并的优化：
1.合并优化
在合并元素x和y时先搜到它们的根结点，然后再合并这两个根结点，即把一个根结点的集改成另一个根结点，这两个根结点的高度不同，如果把高度较小的集合并到较大的集上，能减少树的高度，下面是优化后的代码，在初始化时用h[]定义元素i的高度，在合并时更改。

void union_set(int x,int y)     //合并优化
{
    x=find_set(x);
    y=find_set(y);
    if(h[x]==h[y])
    {
        h[x]++;
        s[y]=x;
    }
    else
    {
        if(h[x]<h[y])s[x]=y;
        else s[y]=x;
    }
}

2.查询优化：
递归

int find1(int x)    //递归
{
    if(x!=s[x])s[x]=find1(s[x]);
    return s[x];
}

非递归

int find2(int x)    //非递归
{
    int r=x;
    while(s[r]!=r)r=s[r];
    int i=x,j;
    while(i!=r)
    {
        j=s[i];
        s[i]=r;
        i=j;
    }
    return r;
}

总代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1050;
int s[maxn],h[maxn];
void init_set()
{
    for(int i=1;i<=maxn;i++)
    {
        s[i]=i;
        h[i]=0;
    }
}
int find_set(int x)
{
    return (x==s[x])?x:find_set(s[x]);
}
int find1(int x)    //递归
{
    if(x!=s[x])s[x]=find1(s[x]);
    return s[x];
}
int find2(int x)    //非递归
{
    int r=x;
    while(s[r]!=r)r=s[r];
    int i=x,j;
    while(i!=r)
    {
        j=s[i];
        s[i]=r;
        i=j;
    }
    return r;
}
void union_set(int x,int y)     //合并优化
{
    x=find1(x);
    y=find1(y);
    if(h[x]==h[y])
    {
        h[x]++;
        s[y]=x;
    }
    else
    {
        if(h[x]<h[y])s[x]=y;
        else s[y]=x;
    }
}

int main()
{
    int  n,t,m,x,y;
    cin>>t;
    while(t--)
    {
        cin>>n>>m;
        init_set();
        for(int i=1;i<=m;i++)
        {
            cin>>x>>y;
            union_set(x,y);
        }
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            if(s[i]==i)
                ans++;

        }
        cout<<ans<<endl;
    }
    return 0;
}