基本思想
用树结构实现并查集的算法思路可以如下描述,假设现在要添加多个路径(点对):
1、初始化:每个点看做一棵树,当然这是一棵只有根节点的树,存储了这个节点本身的值作为组别(你也可以令其他不会产生冲突的记号做组别);
2、查询:对于点对(a,b),通过a和b向其根节点回溯(当然初始时就是它们本身),判断其所在组别;
3、合并:若不在同一组别,令其中一个点(比如a吧)所在树的根节点成为另一个点(比如b)的根节点的孩子。这样即便再查询到a,通过上面的查询过程,程序也会最终判断得到的是现在b的根节点所在的组别,相当于是改变了a所在组的全部元素的组别;
路径压缩
再看看树的生成,这一点就要在查询过程中做文章了,因为每次我们是从树的一个节点回溯到其根节点,所以一个最直接的办法是,将这条路径上的所有中间节点记录下来,全部变成根节点的子节点。但是这样一来会增加算法的空间复杂度(反复开辟内存和销毁)。所以一个备选的思路是每遍历到一个节点,就将这个节点变成他的爷爷节点的孩子(和其父节点在同一层了)。相当于是压缩了查询的路径,这样,频繁的查询当然会导致树的“扁平化”程度更彻底。
基本模板
(1)朴素并查集:
int p[N]; //存储每个点的祖宗节点(父节点)
// 返回x的祖宗节点
int find(int x)
{
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
// 初始化,假定节点编号是1~n
for (int i = 1; i <= n; i ++ ) p[i] = i;
// 合并a和b所在的两个集合:
p[find(a)] = find(b);
(2)维护size的并查集:
int p[N], size[N];
//p[]存储每个点的祖宗节点, size[]只有祖宗节点的有意义,表示祖宗节点所在集合中的点的数量
// 返回x的祖宗节点
int find(int x)
{
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
// 初始化,假定节点编号是1~n
for (int i = 1; i <= n; i ++ )
{
p[i] = i;
size[i] = 1;
}
// 合并a和b所在的两个集合:
size[find(b)] += size[find(a)];
p[find(a)] = find(b);
(3)维护到祖宗节点距离的并查集:
int p[N], d[N];
//p[]存储每个点的祖宗节点, d[x]存储x到p[x]的距离
// 返回x的祖宗节点
int find(int x)
{
if (p[x] != x)
{
int u = find(p[x]);
d[x] += d[p[x]];
p[x] = u;
}
return p[x];
}
// 初始化,假定节点编号是1~n
for (int i = 1; i <= n; i ++ )
{
p[i] = i;
d[i] = 0;
}
// 合并a和b所在的两个集合:
p[find(a)] = find(b);
d[find(a)] = distance; // 根据具体问题,初始化find(a)的偏移量
作者:yxc
链接:https://www.acwing.com/blog/content/404/
来源:AcWing
经典例题
1、AcWing 836. 合并集合
一共有n个数,编号是1~n,最开始每个数各自在一个集合中。
现在要进行m个操作,操作共有两种:
“M a b”,将编号为a和b的两个数所在的集合合并,如果两个数已经在同一个集合中,则忽略这个操作;
“Q a b”,询问编号为a和b的两个数是否在同一个集合中;
输入格式
第一行输入整数n和m。
接下来m行,每行包含一个操作指令,指令为“M a b”或“Q a b”中的一种。
输出格式
对于每个询问指令”Q a b”,都要输出一个结果,如果a和b在同一集合内,则输出“Yes”,否则输出“No”。
每个结果占一行。
数据范围
1≤n,m≤105
输入样例:
4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4
输出样例:
Yes
No
Yes
#include
using namespace std;
const int N = 100010;
int p[N];//存储每个节点的父节点
int find(int x)//返回x的祖宗节点+ 路径压缩
{
if (p[x] != x) p[x] = find(p[x]);//如果p[x]不为根节点,继续递归,直到找到根节点
return p[x];
}
int main()
{
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++ ) p[i] = i;//初始化每个节点
while (m -- )
{
char op[2];
int a, b;
scanf("%s%d%d", op, &a, &b);
if (*op == 'M') p[find(a)] = find(b);//将编号为a和b的两个数所在的集合合并,find(a) 为a的祖宗节点,让a的祖宗节点的父亲等于b的祖宗节点
else
{
if (find(a) == find(b)) puts("Yes");//若两节点的祖宗节点相同 说明两节点在同一集合当中
else puts("No");
}
}
return 0;
}
2、AcWing 837. 连通块中点的数量
给定一个包含n个点(编号为1~n)的无向图,初始时图中没有边。
现在要进行m个操作,操作共有三种:
“C a b”,在点a和点b之间连一条边,a和b可能相等;
“Q1 a b”,询问点a和点b是否在同一个连通块中,a和b可能相等;
“Q2 a”,询问点a所在连通块中点的数量;
输入格式
第一行输入整数n和m。
接下来m行,每行包含一个操作指令,指令为“C a b”,“Q1 a b”或“Q2 a”中的一种。
输出格式
对于每个询问指令”Q1 a b”,如果a和b在同一个连通块中,则输出“Yes”,否则输出“No”。
对于每个询问指令“Q2 a”,输出一个整数表示点a所在连通块中点的数量
每个结果占一行。
数据范围
1≤n,m≤105
输入样例:
5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5
输出样例:
Yes
2
3
注意
1、cnt只有祖节点的有意义
2、要特别注意所有处理cnt的地方,都要“归根结底”
3、在点a和点b之间连一条边时,注意判断a,b 是否在同一集合内
#include
using namespace std;
const int N = 100010;
int n, m;
int p[N], cnt[N];
int find(int x)//寻找x的根节点+路径压缩
{
if (p[x] != x) p[x] = find(p[x]);
return p[x];
}
int main()
{
cin >> n >> m;
for (int i = 1; i <= n; i ++ )//初始化,每个节点刚开始的个数都为1
{
p[i] = i;
cnt[i] = 1;
}
while (m -- )
{
string op;
int a, b;
cin >> op;
if (op == "C")//合并连通块,注意判断a,b 是否在同一集合中,若不在则将a存入b中
{
cin >> a >> b;
a = find(a), b = find(b);
if (a != b)//判断两个节点是否在同一个集合中,即是否为同一个祖先
{
p[a] = b;
cnt[b] += cnt[a];//更新合并后的连通块的点的个数
}
}
else if (op == "Q1")
{
cin >> a >> b;
if (find(a) == find(b)) puts("Yes");
else puts("No");
}
else
{
cin >> a;
cout << cnt[find(a)] << endl;
}
}
return 0;
}
作者:yxc
链接:https://www.acwing.com/activity/content/code/content/45295/
来源:AcWing
3、AcWing 240. 食物链
动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。
A吃B, B吃C,C吃A。
现有N个动物,以1-N编号。
每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述:
第一种说法是”1 X Y”,表示X和Y是同类。
第二种说法是”2 X Y”,表示X吃Y。
此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。
当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
1) 当前的话与前面的某些真的话冲突,就是假话;
2) 当前的话中X或Y比N大,就是假话;
3) 当前的话表示X吃X,就是假话。
你的任务是根据给定的N和K句话,输出假话的总数。
输入格式
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
输出格式
只有一个整数,表示假话的数目。
数据范围
1≤N≤50000,
0≤K≤100000
输入样例:
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
输出样例:
3
思路
采用并査集,但是维护多一个数组 d,用来计算当前节点到父节点的 距离,初始化为 0(因为自身到自身的距离为 0)。根据当前节点到 root 节点的距离(root 节点只能存在一个,其他节点都是 root 节点的子节点)。这个距离算出来只能出现三种情况:0, 1, 2(距离大于或者等于3,取模处理)
//当动物x和动物y的距离%3等于1时,说明x捕食y
//当动物x和动物y的距离%3等于2时,说明y捕食x 也可以说y是x的天敌
//当动物x和动物y的距离%3等于0时,说明x和y是同类
#include
using namespace std;
const int N = 50010;
int n, m;
int p[N], d[N];//d[N]表示节点到根节点的距离,初始值为0
int find(int x)
{
//维护并查集
if (p[x] != x)//如果x不是树根
{
int t = find(p[x]);
d[x] += d[p[x]];
p[x] = t;
}
return p[x];
}
int main()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++ ) p[i] = i;
int res = 0;
while (m -- )
{
int t, x, y;
scanf("%d%d%d", &t, &x, &y);
if (x > n || y > n) res ++ ;//res记录假话的个数
else
{
int px = find(x), py = find(y);
if (t == 1)//真话 x和y是同类
{
if (px == py && (d[x] - d[y]) % 3) res ++ ;//如果x, y在同一集合内,x到根节点距离减去y到根节点距离为0,则x,y为同类,否则该句话为假
else if (px != py)//如果x,y不在同一个集合内,把x插入y中
{
p[px] = py;
d[px] = d[y] - d[x];/*因为合并x和y所在集合多出了一段长度
这块长度是find(x)到find(y)的距离 所以求多出来的这块部分的长度
当x和y是同类时,有这样的特性(d[x]+d[px]-d[y])%3
这里的d[x]是还未合并时,x到px的距离
∴d[px]=d[y]-d[x]
*/
}
}
else//真话 x捕食y
{
/*当x和y在一个集合中时,由题目可知,x捕食y
此时有x到根节点的距离-y到根节点的距离=1+3k k为任意实数
∴当(d[x]-d[y]-1-3k)%3 ==0 时可确认 x捕食y
反之当(d[x]-d[y]-1-3k)%3 !=0 x不可能捕食y */
if (px == py && (d[x] - d[y] - 1) % 3) res ++ ;
else if (px != py)//当x和y不在一个集合时,将x和y所在集合合并
{
/*
设find(x)到find(y)的距离为d([find(x)])
此时有d[x]+d([find(x)])-d[y]=3k+1
∴d[find(x)]=-d[x]+d[y]+1+3k
*/
p[px] = py;
d[px] = d[y] + 1 - d[x];
}
}
}
}
printf("%d\n", res);
return 0;
}
作者:yxc
链接:https://www.acwing.com/activity/content/code/content/45325/
来源:AcWing