并查集（学习笔记）

并查集（学习笔记）

并查集是一种用来管理元素分组情况的数据结构
整体上分为两个部分：
合并集合：合并元素a和元素b所在的组
查找集合：查询元素a和元素b是否属于同一组

模板：

/*初始化：
	增添一个parent数组，存储每个元素对应的根结点*/
	for(int i = 0; i < MX; i++) parent[i] = i; //初始化 
//查找：
	int findset(int x)
	{ 
	    if(parent[x] == x) return x; 
	    int rt = findset(parent[x]);
  	    parent[x] = rt; 
 	    return rt; 
	}
	/*注：为了节省内存，可以把省去rt，返回时直接返回parent[x] */
	//简洁版
	int findset(int x) 
	{ 
          return parent[x] == x ? parent[x] = x:findset(parent[x]); 
	}
//合并
	void merger(int le, int ri) 
	{ 
	    int l = findset(le); 
	    int r = findset(ri); 
	    parent[l] = r; 
	}

应用：
判环；分组（即最后形成了几棵树）

例题：1. HDU 1232 ---- 畅通工程

Problem Description
某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？

Input
测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数，分别是城镇数目N ( < 1000 )和道路数目M；随后的M行对应M条道路，每行给出一对正整数，分别是该条道路直接连通的两个城镇的编号。为简单起见，城镇从1到N编号。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时，输入结束，该用例不被处理。

Output
对每个测试用例，在1行里输出最少还需要建设的道路数目。

Sample Input

4 2
1 3
4 3
3 3
1 2
1 3
2 3
5 2
1 2
3 5
999 0
0

Sample Output

1
0
2
998

Hint

Huge input, scanf is recommended.

#include <iostream>
#include <cstdio>
using namespace std;
const int MAX=1010;
int parent[MAX]; //父结点数组，存放各个结点的父结点是几号结点
int ans;

/* 为了不超出内存限制，findroot函数里面的x要换成parent[x]，中间变量root也要省去*/
/*int findroot(int x)
{   这样写会超内存
    if (parent[x]==x) return x;
    int root=findroot(x);
    parent[x]=root;
    return root;
}
*/

int findroot(int x)
{   //查找根结点
    if (parent[x]==x) return parent[x]; //当前结点的根结点是自身，说明当前点为根结点
    parent[x]=findroot(parent[x]);   //当前不是根结点，继续递归查找
    /*parent[x]的值经过层层递归，最终变成根结点的号码，回退时，
      作为父结点依次传给路径上的每一个子节点 */
    return parent[x];
    /*例如：有路径4->3->1->2，2是4,3,1的根结点，是1的父结点。那么parent[1]
        储存的就是根结点的号码----2，即1号的父结点是2号，
        最后返回parent[1]；返回parent[1]到x=3的这层函数中，然后把parent[1]
        （值为2）赋给parent[3],即3号的父结点是1号，最后接着返回parent[3]，
        进入x=4的那层函数......*/
}
/*简洁版  这个没试过，不知道内存超不超
	int findroot(int x)
	{
          return parent[x] == x ? parent[x] = x:findroot(parent[x]);
	}
*/

void Merge(int lef,int rig)
{   //合并函数
    //传入的左边结点的根结点，要与传入的右结点的根结点合并，即前者的父结点是后者
    //此操作为通用操作，可以处理两个单独的结点，也可以处理两棵树
    int l=findroot(lef);
    int r=findroot(rig);
    if (l==r) return; //左边和右边的结点已经在一个集合中，不作处理
    parent[l]=r;  //令右边的结点为左边结点的父结点，两棵树（两个结点）成功连接
    ans--; //连接成功，不用再建这条路，-1
}

int main()
{
    int n,m;
    int a,b;
    while (~scanf("%d",&n))
    {
        if(n==0) break;
        scanf("%d",&m);
        for (int i=0 ; i<=n ; i++)
            parent[i]=i;
        //上述语句为初始化操作，先让各个结点指向自己，即每个结点的父结点为自身
        //例如：0号的父结点是0号，1号的父结点是1号......
        ans=n-1; // 目前没有道路，则需要建n-1条
        while(m--)
        {
            scanf("%d%d",&a,&b);
            //让当前互通的两个村庄合并
            Merge(a,b);
        }
        //parent数组中有几个结点对应的父结点是自身，就有几棵树

        //这里之前用的是i=0 ; i<n  这样是不行的，因为村庄是从1开始排的，而不是0
        //for (int i=1 ; i<=n ; i++)
        //    if (parent[i]==i) ans++;
        printf("%d\n",ans);
    }
    return 0;
}