并查集的优化
并查集其实不是一个很难的知识点,之前也是看会了普通版就知足了。直到这次比赛,并查集的题目超时了才让我明白精益求精的重要性,所以回来总结并查集的优化操作。
首先来复习一下并查集的基本操作
初始化
void init()
{
for(int i=1;i<=n;i++)
fa[i]=i;
}
找爹函数
int find(int x)
{
while(x!=fa[x])
x=fa[x];
return x;
}
合并函数
void unite(int x,int y)
{
x=find(x);
y=find(y);
if(x!=y)
fa[x]=y;//不论树高直接莽上去连接
}
这样写确实没什么大问题,我之前就是这么干的,但是这样写容易超时,尤其是在比赛的时候,不会优化真的当场瓦掉,不能输在简单题上面!
路径压缩
int find(int x){
if(x==fa[x]){
return x;
}else return fa[x] = find(fa[x]);/*直接做祖先的儿子,
避免出现一长条的情况,以此达到减少时间复杂度的目的*/
}
但是这个算法依旧存在着一个很大的问题:只有在查找了上级的元素才能进行压缩,并且第一次执行时,是不能达到压缩效果的。只有在之后才能有效。
这样可能就会出现原本属于一个集合的元素,在简单运作一遍之后却并没有归到一个集合里面,只有在路径压缩一遍才能达到想要的结果。
我之前是这样解决的,学长说我很不规范。。。
for(int i=1;i<=n;i++)
find(i);
但是后面这个优化可以完美解决这个问题
按秩合并
这个优化的原理其实蛮好懂的,就是把一个个集合做成树状,在合并的时候把比较矮的树连接到比较高的树下面,从而达到查询时减少时间复杂度的目的。(这个优化可以非常神奇的解决上面的那个问题,具体怎么怎么解决的,跟着程序人脑调试一波就能明白了\狗头)
这里需要先建立一个树高数组来记录树高,初始化时每个元素的树高都是1,
后面更改的时候只需要更改集合代表元素的树高即可。
void unite(int x,int y)
{
x=find(x);
y=find(y);
if(x==y)
return ;
if(rk[x]<rk[y])//rk[]数组是树高的意思,不断选取树高更矮的作为根节点连接
fa[x]=y;
else
{fa[y]=x;
if(rk[x]==rk[y])
rk[x]++;//树高相同时合并一定会增加树高
}}
好家伙,会了这些,来做个题吧:
题目描述:广播系统
为了更加快速的传递学习任务,ACM集训队计划建设一个广播系统!按照规划,这个系统包含若干端点,这些端点由神奇的网络连接。
此网络有下述特点:
1.消息可以在任何一个端点产生,并且只能通过这个网络传递信息。每个端点接收消息后会将消息传送到与其相连的端点(单项传输,不会传输到那个消息发送过来的端点)
2.如果某个端点是产生消息的端点,那么消息将被传送到与其相连的每一个端点。
3.当消息在某个端点生成后,其余各个端点均能接收到消息
4.任意一个消息可以被快速的传给所有端点
现给你这个广播系统的连接方案,你能判断此系统是否符合以上要求并且传递给所有的端点?
输入:
输入包含多组测试数据。每两组输入数据之间由空行分隔。
每组输入首先包含2个整数N和M,N(1<=N<=1000)表示端点个数,M(0<=M<=N*(N-1)/2)表示通信线路个数。
接下来M行每行输入2个整数A和B(1<=A,B<=N),表示端点A和B由神奇的网络相连。两个端点之间至多由一条网络直接相连,并且没有将某个端点与其自己相连的网络。
当N和M都为0时,输入结束。
输出
对于每组输入,如果所给的系统描述符合题目要求,则输出Yes,否则输出No。
样例输入
4 3
1 2
2 3
3 4
3 1
2 3
0 0
样例输出
Yes
No
解题思路
这题就是很基础的并查集,相信会并查集的各位都能很容易的写出来。根据题目中的特点1,我们发现这个网络是单向传输的,也就是如果是甲发的消息,那么甲就不能接受到这条消息。换句话说,这个系统不能有环状结构存在。所以我们不光需要判断根节点是不是只有一个,还需要判断是不是有环。受到大佬(博客名:CTGU-liyb)的点拨,发现判断是不是环,只需要判断端点的数目是不是等于线路的数目加一即可。
我给出一版能够AC的代码,我上面说用传统方法会超时也是说的这道题。
#include<iostream>
using namespace std;
int fa[1000+10];//父亲数组
int r[1000+10];//树数组
int n, m;
int find(int x){
if(x == fa[x]){
return x;
}else {
return fa[x] = find(fa[x]);
}
}
void unite(int x, int y){
int a = find(x);
int b = find(y);
if(a == b) return ;
if(r[a] > r[b]){
fa[b] = a;
}else if(r[b] > r[a]){
fa[a] = b;
}else if(r[a] == r[b]){
fa[b] = a;
r[a]++;
}
}
int main()
{
while(scanf("%d%d", &n, &m) != EOF){
if(n == 0 && m == 0) break;
for(int i=1;i<=n;i++){
fa[i] = i;
r[i] = 1;
}
int a, b;
for(int i=1;i<=m;i++){
scanf("%d%d", &a, &b);
unite(a,b);
}
int flag = fa[1];
for(int i=1;i<=n;i++){
if(fa[i] != flag){
flag=0;
break;
}
}
if(flag==0 || m!=(n-1)) printf("No\n");
else printf("Yes\n");
}
return 0;
}