并查集算法

擒贼先擒王

#include <iostream>
#include <fstream>
#include <cstring>
#include <vector>
#include <queue>
#include <stack>
#include <algorithm>
#include <cmath>

using namespace std;

#define loop(i,n) for(int i=0;i<(n);i++)
#define loop2(i,n) for(int i=1;i<=(n);i++)

const int maxn=1000;
int inf=99999999;

int f[maxn];
int n,m,k,sum=0;
//这是初始化，非常重要，数组里面存的是自己数组下标的编号就好了
void init()
{
  for(int i=1;i<=n;i++)
    f[i]=i;
}
//这是找爹的递归函数，不停地去找爹，直到找到祖宗为止，其实就是去找犯罪团伙的最高领导人，擒贼先擒王原则
int getf(int v)
{
  if(f[v]==v)
    return v;
  else
  { //路径压缩，每次在函数返回的时候，顺带把路上遇到的人的BOSS改为最后找到的祖宗编号，也就是犯罪团伙的最高领导人编号
    f[v]=getf(f[v]);
    return f[v];
  }
}
//这里是合并两子集的函数
void merge(int v,int u)
{
  int t1,t2;
  t1=getf(v);
  t2=getf(u);
  if(t1!=t2) //判断两个结点是否在同一个集合中，即是否为同一个祖先
    f[t2]=t1; //"靠左“原则，左边变成右边的BOSS。即把右边的集合，作为左边集合的子集合，经过路径压缩以后，将f[u]的根的值也赋值为v的祖先f[t1]
}

void test()
{   
  freopen("unionn.in","r",stdin);  
  //freopen("unionn.out","w",stdout); 
  int x,y;
  cin>>n>>m;
  init();
  for(int i=1;i<=m;i++)
  {
    cin>>x>>y;
    merge(x,y); //开始合并犯罪团伙   
  }

  for(int i=1;i<=n;i++)  //最后扫描有多少个独立的犯罪团伙
  {
    cout<<f[i]<<" ";
    if(f[i]==i)
      sum++;
  }
  cout<<endl;
  cout<<sum<<endl; 
}

int main () 
{        
    test();        
    return 0;
}

test data:

10 9
1 2
3 4
5 2
4 6
2 6
8 7
9 7
1 6
2 4