并查集用于集合合并的一种数据结构
原理
它的原理是最开始的时候把所有集合的元素都初始化为-1。所有元素中值为负的元素,都代表一个小集合的根元素。合并的时候,如果要将俩个子元素合并,则要把俩个子元素的所在集合合并,则先找到这俩个元素的根,然后把它们所代表的元素值累加到另一个当中,然后更新另一个的下标为被累加元素的下标,从此之后俩个元素的所在的俩个集合得到了合并。
这样合并后,之前另一个根的子元素,上溯寻找根的时候,找到原来的根后,继续上溯就会找到新的根。
代码
#include<iostream>
#include<vector>
/*并查集*/
using namespace std;
class Uoion
{
public:
Uoion(int n=0)
:_uoion(n+1,-1)
{}
int uni(int x ,int y)
{
int Root = x ;
int Root2 = y;
while(_uoion[Root]>=0||_uoion[Root2]>=0)
{
if(_uoion[Root]>=0)
Root = _uoion[Root];
if(_uoion[Root2]>=0)
Root2 = _uoion[Root2];
}
if(Root!=Root2)
{
_uoion[Root]+=_uoion[Root2];
_uoion[Root2] = Root;
}
return Root;
}
int Size()
{
int ret = 0;
size_t size = _uoion.size();
for(size_t i=0 ; i<size ; ++i)
{
if(_uoion[i]<0)
++ret;
}
return ret;
}
private:
vector<int> _uoion;
};
int friends(int n , int m , int r[][2])//int (*m) [2]
{
Uoion object(n);
for(int i=0;i<m;++i)
{
object.uni(r[i][0],r[i][1]);
}
return object.Size()-1;
}
int main()
{
int n = 5;
int m = 3;
int r[m][2] = {{1,2},{2,3},{4,5}};
cout<<friends(n,m,r)<<endl;
return 0;
}