并查集反集
食用指南:
对该算法程序编写以及踩坑点很熟悉的同学可以直接跳转到代码模板查看完整代码
只有基础算法的题目会有关于该算法的原理,实现步骤,代码注意点,代码模板,代码误区的讲解
非基础算法的题目侧重题目分析,代码实现,以及必要的代码理解误区
题目描述:
-
现在有 n 个人,他们之间有两种关系:朋友和敌人。我们知道:
一个人的朋友的朋友是朋友
一个人的敌人的敌人是朋友
现在要对这些人进行组团。两个人在一个团体内当且仅当这两个人是朋友。请求出这些人中最多可能有的团体数。输入格式
第一行输入一个整数 n 代表人数。
第二行输入一个整数 m 表示接下来要列出 m 个关系。
接下来 m 行,每行一个字符 opt 和两个整数 p,q,分别代表关系(朋友或敌人),有关系的两个人之中的第一个人和第二个人。其中 optopt 有两种可能:如果 opt 为 F,则表明 pp 和 qq 是朋友。
如果 opt 为 E,则表明 pp 和 qq 是敌人。输出格式
一行一个整数代表最多的团体数。输入输出样例
输入
6
4
E 1 4
F 3 5
F 4 6
E 1 2
输出
32 ≤ n ≤ 1000
1 ≤ m ≤ 5000
1 ≤ p,q ≤ n -
题目来源:https://www.luogu.com.cn/problem/P1892
题目分析:
- 集合问题,必然采用并查集
- 求集合最大数目,必然采用count[N]数组
- 两种关系:
若x y为朋友,则x y所属集合直接合并
若x y为敌人,则x所属集合 和 y任一敌人所属集合合并 - 集合合并时,并不是直接将x所属集合和 y所属集合进行合并
而是可能将x集合和y所属对立集合合并
或将y集合和x所属对立集合合并
这样的并查集集合合并问题称为并查集反集 - 下面开始讲解并查集反集
顺便看看反集题目中的集合内点数如何统计(本题没要求)
算法原理:
模板算法:
- 传送门:并查集
- 还对并查集不是很熟悉的同学一定要先看完并查集内容
合并时带有判断的并查集问题:
1. 流程分析:
- 当x y为朋友:
直接 uni(find(x), find(y)) - 当x y为敌人:
当y此时没有敌人:将x设置为y的敌人
当y已经有敌人时:uni(find(x), find(y的敌人))
2. 存储结构:
- 敌人数组 enemy[N];
- 由分析可知,我们要为每个元素建立一个敌人数组,存放这个元素的一个敌人元素即可
- 当其他敌人和该元素敌人数组中的已有敌人进行合并时,最终产生的也是中心结构的连通集合
- 最大集合内点数:
- 法一:由于n最大才1000,最后遍历一次count[N]即可
- 法二:每次集合合并时候就将maxx和合并后的count[中心点]比较
代码实现:
- 虽然本题没要求求出集合内最多点数,但是多写当练习
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1010;
int n;
int fa[N];
int cnt[N];
int enemy[N];
int maxx;
void init(){
for(int i=1; i<=n; i++) fa[i]=i, cnt[i]=1;
}
int find(int x){
if (fa[x] == x) return x;
return fa[x] = find(fa[x]);
}
void uni(int x, int y){
int fax = find(x);
int fay = find(y);
if (fax < fay){
fa[fay] = fax;
cnt[fax] += cnt[fay];
maxx = max(maxx, cnt[fax]);
}else if(fax > fay){
fa[fax] = fay;
cnt[fay] += cnt[fax];
maxx = max(maxx, cnt[fay]);
}
}
int main(){
int m = 0;
cin >>n >>m;
init();
while(m--){
string s;
int x=0, y=0;
cin >>s >>x >>y;
if(s == "F") uni(x, y);
else{
if(!enemy[y]) enemy[y] = x;
else uni(x, enemy[y]);
if(!enemy[x]) enemy[x] = y;
else uni(y, enemy[x]);
}
}
int ans = 0;
for(int i=1; i<=n; i++)
if(fa[i] == i)
ans++;
cout <<ans;
return 0;
}
代码误区:
1. 为什么每个元素的敌人只需要保存一个?
-
当该点本来无敌人时,现在刚出现的敌人集合不存在和谁合并
-
当该点已经有了一个敌人时,
若之后出现的是原本的敌人集合,uni()两个相同集合,count[N]不变若之后出现多个不同的敌人集合,则不同集合进行合并也只有一个中心点,依据中心点又可以判断该区间是否已经合并过了
2. 在反集进行合并时,为什么是双向的?
-
现在x和y是敌人关系
- 将x所属集合和enemy[y]所属集合进行合并
- 将y所属集合和enemy[x]所属集合进行合并
-
下一次又出现z 和 x 是集合关系呢?
上述的 1 & 2若少了第2条
此时的z集合极可能就和y集合永远连接不上同理,下次是z 和 y是集合关系时,缺少1也可能导致z 和 x永远连接不上
本篇感想:
- 本题的难点就在对立数组的建立,
想到这一点,结合我的代码误区第二问,这个题已经做完了
我刚刚做的时候写了个init(),结果TM忘记调用了,这波给我记牢了 - 本题摘自洛谷 绿色的普及+/提高,
还记的我大一上学期做洛谷第一题,真戁,
现在已经可以独立分析和写作绿色组题目了,开心 - 看完本篇博客,恭喜已登 《练气境-后期》
距离登仙境不远了,加油