一、并查集模板
并查集是一种树型的数据结构,用于处理一些不交集(Disjoint Sets)的合并及查询并划分的问题。有一个联合-查找算法(union-find algorithm)定义了两个用于此数据结构的操作:
Find:确定元素属于哪一个子集。它可以被用来确定两个元素是否属于同一子集。
merge:将两个子集合并成同一个集合。
void init()
{
for(int i=1;i<=n;i++)
fa[i]=i;//对n个点进行根节点初始化
r[i]=1;//大小初始化
]
int find(int x)//寻找x的根节点
{
if(x!=fa[x[) fa[x]=find(fa[x])// 将父节点也直接和根节点相连,降低树的深度
return fa[x];
}
void merge(int x,int y)
{
int a = find(x);
int b = find(y);
if(a!=b)//根节点打架,选择一方屈服
{
if(r[a]<r[b]) fa[a]=b,r[b]+=r[a];
else fa[b]=a,r[a]+=r[b];//按大小合并,还有一种是按树的深度合并
}
}
二、例题
1.交际圈
题目分析:先判断合并后根节点是否唯一,若不唯一求出最大规模和最小规模,相减
代码如下(示例):
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 1e6;
int fa[N];
int cnt[N];
int find(int x)
{
return x == fa[x]? x : fa[x] = find(fa[x]);
}
void merge(int x ,int y)
{
int a = find(x);
int b = find(y);
if(a != b)
{
if(cnt[a]<cnt[b]) fa[a]=b,cnt[b]+=cnt[a];
else fa[b]=a,cnt[a]+=cnt[b];
}
}
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
fa[i] = i;
cnt[i] = 1;
}
for(int i=1;i<=m;i++)
{
int x,y;
cin>>x>>y;
merge(x,y);
}
int maxn = -1,minn = 9999999;
for(int i=1;i<=n;i++)
{
if(fa[i] == i)
{
maxn = max(maxn,cnt[i]);
minn = min(minn,cnt[i]);
}
}
if (maxn == n) printf("WOW\n");
else printf("%d\n",maxn-minn);
return 0;
}
2.广播系统
题目描述:
为了更加快速的传递学习任务,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。
样例输入 Copy
4 3
1 2
2 3
3 4
3 1
2 3
0 0
题目分析:输出yes条件就是根节点唯一,并且不能成环
代码如下(示例):
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 1e6;
int fa[N],r[N];
int n,m;
void init()
{
for(int i=1;i<=n;i++)
fa[i]=i,r[i]=1;
}
int find(int x)
{
if(x!=fa[x]) fa[x]=find(fa[x]);
return fa[x];
}
void merge(int x,int y)
{
int a = find(x);
int b = find(y);
if(a!=b)
{
if(r[a]<r[b]) fa[a]=b,r[b]+=r[a];
else fa[b]=a,r[a]+=r[b];
}
}
int main()
{
while(cin>>n&&n)
{
cin>>m;
init();
for(int i=0;i<m;i++)
{
int x,y;
cin>>x>>y;
merge(x,y);
}
int maxn = 0;
for(int i=1;i<=n;i++)
{
if(fa[i]==i)
maxn = max(maxn,r[i]);
}
if(maxn==n&&(m==n-1)) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
return 0;
}