大学的同学来自全国各地,对于远离家乡步入陌生大学校园的大一新生来说,碰到老乡是多么激动的一件事,于是大家都热衷于问身边的同学是否与自己同乡,来自新疆的小赛尤其热衷。但是大家都不告诉小赛他们来自哪里,只是说与谁同乡,从所给的信息中,你能告诉小赛有多少人确定是她的同乡吗?
输入
每个测试实例首先包括2个整数,N(1 <= N <= 1000),M(0 <= M <= N*(N-1)/2),代表现有N个人(用1~N编号)和M组关系;
在接下来的M行里,每行包括2个整数,a,b,代表a跟b是同乡;
当N = 0,M = 0输入结束;
已知1表示小赛本人。
样例输入
3 1
2 3
5 4
1 2
3 4
2 5
3 2
0 0
输出
对于每个测试实例,输出一个整数,代表确定是小赛同乡的人数。
样例输出
0
4
这道题是并查集的应用
#include <iostream>//并查集简单应用
using namespace std;
#define N 1001
int n,m;
int pa[N];
void makeset(int n)//初始化并查集
{
for(int i=1;i<=n;++i)
pa[i]=i;
}
int findset(int x)//查找x所在集合的根节点
{
if(x!=pa[x])
return findset(pa[x]);
return pa[x];
}
void unionset(int x,int y)//两个集合合并
{
x = findset(x);
y = findset(y);
if(x == y)
return ;
pa[x] = y;
}
int main()
{
int n,m,a,b,i;
while(cin>>n>>m)
{
if(n==0&&m==0)
break;
makeset(n);
for(int i=0;i<m;++i)
{
cin>>a>>b;
a=findset(a);
b=findset(b);
unionset(a,b);
}
int cnt=0;
int result=findset(1);
for(int i=2;i<=n;++i)//查找和1的根节点相同的节点
{
if(findset(i)==result)
++cnt;
}
cout << cnt<< endl;
}
return 0;
}
再来复习一下并查集的用法
//建立一个新的集合,每一个子节点就是一个数,本身就是他的根节点
void Make_Set(int x)
{
father[x] = x;
R[x] = 0;
}
//返回能代表x所在集合的节点,通常返回x所在集合的根节点
int Find_Set(int x)
{
if(x != father[x])
father[x] = Find_set(father[x]);
return father[x];
}
//两个集合的合并算法
//合并时,如果两个集合的秩相同,任选一个根做为父节点,并增加其秩。
//秩不同时,让较小秩的集合指向较大秩的集合,这时秩的大小不变。
void Union(int x, int y)
{
int GrandX = Find_set(x);
int GrandY = Find_set(y);
if(GrandX == GrandY)
return;
if(R[GrandX] < R[GrandY])
father[GrandX] = GrandY;
else
{
if(R[GrandX] == R[GrandY])
R[GrandX]++;
father[GrandY] = GrandX;
}
}