题意:有n名学生m个团体,0号学生疑似得了传染病(SARS),所以和0号学生一个团体的人都是嫌疑者,要得出这个团体有多少个人。
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define N 40000
int pre[N],sum[N];//记录作用
int find(int x)
{
if(x!=pre[x])
{
pre[x]=find(pre[x]);
}
return pre[x];
}
int main()
{
int n,m;
while(cin>>n>>m&&(n||m))
{
for(int i=0;i<n;i++)
{
pre[i]=i;
sum[i]=1;
}
while(m--)
{
int t,num,one;
cin>>t>>one;
t--;
while(t--)
{ cin>>num;
int bosx=find(one);
int bosy=find(num);
if(bosx!=bosy)
{
pre[bosy]=bosx;
sum[bosx]+=sum[bosy];
}
}
}
cout<<sum[find(0)]<<endl;
}
}
最初每个人都是一个集合,所以pre[i]=i;每个人的父节点都是他们本身
通过所给的数据,将两个相联系的集合连起来变成一个集合。
剩下没有连在一起的集合就是另一个独立个体
假如 1 2 3 4 5 6
1 和 2 是一个集合后,如果 1 和3也是一个集合,呢么2 和3 也是一个集合,所以只用一个数和他们相联系就可以
因为最开始每一个集合都是1
所以我们给 sum[i]=1;
因为我们要找和0号一个团队的人是多少,我们就需要找到0所在团队的编号是多少,pre[0],就是他们的编号,因为是一个集合所以他们的编号是相同的,pre[i]=pre[i+1];
sum[x]+=sum[y],每找到是一个集合的人这个队伍就壮大1个,x是这个队伍的编号。
HDU The Suspects
最新推荐文章于 2020-03-11 12:02:47 发布