(题解来自北大2016 ACM/ICPC竞赛训练暑期课讲义)
题目链接:http://poj.org/problem?id=1611
题意: n个学生分属m个团体,(0 < n <= 30000 , 0 <= m <= 500) 一个学生可以属于多个团体。 一个学生疑似患病,则它所属的整个团体都 疑似患病。已知0号学生疑似患病,以及每个 团体都由哪些学生构成,求一共多少个学生 疑似患病。
n个学生分属m个团体,(0 < n <= 30000 , 0 <= m <= 500) 一个学生可以属于多个团体。 一个学生疑似患病,则它所属的整个团体都 疑似患病。已知0号学生疑似患病,以及每个 团体都由哪些学生构成,求一共多少个学生 疑似患病。
解法:最基础的并查集,把所有可疑的都并一 块。
#include<iostream>
using namespace std;
#define maxn 30010
int n,m,k;
int parent[maxn];
int total[maxn];
int GetParent(int a) //获取a的根 把a的父节点改为根
{
if (parent[a]!=a)
parent[a]=GetParent(parent[a]);
return parent[a];
}
void Merge(int a,int b)
{
int p1=GetParent(a);
int p2=GetParent(b);
if (p1==p2)
return;
total[p1]+=total[p2];
parent[p2]=p1;
}
int main()
{
while (true)
{
scanf("%d%d",&n,&m);
if (n==0&&m==0)
break;
for (int i=0;i<n;i++)
{
parent[i]=i;
total[i]=1;
}
for (int i=0;i<m;i++)
{
int h,s;
cin>>k;
cin>>h;;
for (int j=1;j<k;j++)
{
cin>>s;
Merge(h,s);
}
}
printf(“%d\n”,total[GetParent(0)]);
}
return 0;
}