并查集——基础

并查集

并查集

并查集是一种树型的数据结构，用于处理一些不交集（Disjoint Sets）的合并及查询问题。有一个联合-查找算法（union-find algorithm）定义了两个用于此数据结构的操作：

Find：确定元素属于哪一个子集。它可以被用来确定两个元素是否属于同一子集。

Union：将两个子集合并成同一个集合。

模板

void init(){
	for(int i=1;i<=n;i++){
		pre[i]=i;
	}
}
int find(int x){
	if(pre[x]==x)	return x;
	pre[x]=find(pre[x]);
	return pre[x];
}
void unionn(int x,int y){
	int rx=find(x);
	int ry=find(y);
	pre[rx]=ry;
}

例题

POJ-2236 Wireless Network

POJ-2236

题目描述

南亚发生了一次地震。ACM (Asia Cooperated Medical 亚洲联合医疗队) 已经为膝上型电脑搭建了一个无线网络，但受到了一次不可预知的余震攻击，因此网络中的所有电脑都被破坏了。电脑被逐台修复，网络逐步恢复了工作。由于受到硬件的约束，每台电脑只能与距离它不超过 d 米的其它电脑直接通信。但每台电脑可被看作其它两台电脑的通信中转点，也就是说，如果电脑 A 和电脑 B 可以直接通信，或存在一台电脑 C 既可与 A 也可与 B 通信，那么电脑 A 和电脑 B 之间就能够通信。

在处理网络修复的过程中，工作人员们在任何一个时刻，可以执行两种操作：维修一台电脑，或测试两台电脑是否能够通信。请您找出全部的测试操作。

输入

第一行包含了两个整数 N 和 d (1 <= N <= 1001, 0 <= d <= 20000)。此处 N 是电脑的数目，编号从 1 到 N；同时，D 是两台电脑之间能够直接通信的最大距离。接下来的 N 行，每行包含两个整数 xi, yi (0 <= xi, yi <= 10000)，表示 N 台电脑的坐标。从第 (N+1) 行到输入结束，是逐一执行的操作，每行包含一个操作，格式是以下两者之一：
\1. “O p” (1 <= p <= N)，表示维护电脑 p 。
\2. “S p q” (1 <= p, q <= N)，表示测试电脑 p 和 q 是否能够通信。

输入不超过 300000 行。

输出

对于每个测试操作，如果两台电脑能够通信，则打印 “SUCCESS”；否则，打印 “FAIL”。

示例输入

4 1
0 1
0 2
0 3
0 4
O 1
O 2
O 4
S 1 4
O 3
S 1 4

示例输出

FAIL
SUCCESS

思路分析

若两个点已经被维护且距离小于等于d，那么将它们合并

详细代码

#include<cstdio>
#include<iostream>
#include<cmath>
using namespace std;
const int maxn=1002;
int n,d;
int pre[maxn];
int vis[maxn];
struct node{
	int x,y;
}arr[maxn];
void init(){
	for(int i=1;i<=n;i++){
		pre[i]=i;
	}
}
int find(int x){
	if(x==pre[x])	return x;
	pre[x]=find(pre[x]);
	return pre[x];
}
void unionn(int x,int y){
	int rx=find(x);
	int ry=find(y);
	if(rx!=ry){
		pre[rx]=ry;
	}
}
double getDis(node& a,node& b){
	return sqrt((double)(a.x-b.x)*(a.x-b.x)+(double)(a.y-b.y)*(a.y-b.y));
}
int main(){
	cin>>n>>d;
	for(int i=1;i<=n;i++){
		cin>>arr[i].x>>arr[i].y;
	}
	init();
	int x,y,cnt=0;
	while(1){
		char ch='O';
		while(cin>>ch && ch!='S'){
			cin>>x;
			vis[cnt++]=x;
			for(int i=0;i<cnt-1;i++){
				if(getDis(arr[vis[i]],arr[vis[cnt-1]])<=d){
					unionn(vis[i],vis[cnt-1]);
				}
			}
		}
		if(ch=='O')	break;
		cin>>x>>y;
		if(find(x)!=find(y)){
				cout<<"FAIL"<<endl;
		}else{
			cout<<"SUCCESS"<<endl;
		}
		
	}
	return 0;
}

POJ-1611 The Suspects

POJ-1611

题目描述

总所周知，大佬喜欢和大佬交流和组队。已知有多个讨论小组，我们认为如果这个讨论小组里有一个人可能是大佬，那么这个小组所有成员都可能是大佬。我们已经知道第0号同学是大佬，请你帮忙数数有多少人可能是大佬。

注:如果0号同学所在的小组里有1号同学，那么1号同学可能是大佬，1号同学所在的小组里的人也都有可能是大佬，以此类推。

Input

输入文件包含多组数据。

对于每组测试数据:
第一行为两个整数n和m, 其中n是学生的数量, m是讨论小组的数量。 接下来m行，每一行有一个整数k，代表讨论小组中学生的数量。之后，有k个整数代表这个小组里每个 学生的编号(在0到n-1之间)。
n = m = 0表示输入结束，不需要处理。

0 < n ≤30000，0 ≤ m ≤ 500。

Output

对于每组输入，输出一行一个整数表示可能是大佬的人的数量

Sample Input

100 4
2 1 2
5 10 13 11 12 14
2 0 1
2 99 2
200 2
1 5
5 1 2 3 4 5
1 0
0 0

Sample Output

4
1
1

思路分析

并查集模板题目，将每个组的学生合并到同一个集合中。

详细代码

#include<cstdio>
#include<iostream>
using namespace std;
int a[30002];
int n,m;
int pre[30002];

void init(){
	for(int i=0;i<n;i++){
		pre[i]=i;
	}
}
int find(int x){
	if(pre[x]==x)	return x;
	pre[x]=find(pre[x]);
	return pre[x];
	
}
void unionn(int x,int y){
	int rx=find(x);
	int ry=find(y);
	pre[rx]=ry;
}
int main(){
	
	while(cin>>n>>m && (n||m)){
		init();
		int cnt=0;
		for(int i=0;i<m;i++){
			int k;
			scanf("%d",&k); 
			for(int j=0;j<k;j++)	cin>>a[j];
			for(int j=0;j<k;j++){
				unionn(a[0],a[j]);
			}	
		}
	
		for(int i=0;i<n;i++){
			if(find(0)==find(i))	cnt++;
		}
			
		cout<<cnt<<endl;
	}
	
	return 0;
}