题意:编号为0的同学得了病,会传染给所有与他同一小组的人。给定M个小组的成员名单(一个人可以加入多个小组),求多少人生病了。
分析:在同一集合里就会被传染,并查集。
代码:
# include <algorithm>
# include <cstdio>
const int SIZEN=30000+5;
const int SIZEM=500+5;
int n , m , f[SIZEN] ;
int _find ( int ) ;
void _union ( int , int ) ;
int main () {
//freopen ( "SARS.in" , "r" , stdin ) ;
while ( scanf ( "%d%d" , &n , &m ) ) {
if ( n==0 && m==0 ) break ;
if ( m==0 ) { printf ( "1\n" ) ; continue ; }
for ( int i=0 ; i!=n+1 ; ++i ) f[i]=i;
for ( int i=0 ; i!=m ; ++i ) {
int k,tem[30005];
scanf ( "%d" , &k ) ;
for ( int j=0 ; j!=k ; ++j ) {
scanf ( "%d" , &tem[j] ) ;
_union ( tem[j] , tem[0] ) ;
}
}
int tot=0;
int root=_find(0);//0号所在集合的根
for ( int i=0 ; i!=n+1 ; ++i ) tot+=(_find(i)==root);//同一集合则会被传染
printf ( "%d\n" , tot ) ;
}
return 0 ;
}
int _find ( int i ) {
return f[i]==i ? i : ( f[i]=_find(f[i]) ) ;
}
void _union ( int i , int j ) {
int fi=_find(i) ;
int fj=_find(j) ;
if ( fi!=fj ) f[fi] = fj ;
return ;
}