题意:有N个人和M个亲戚关系,规定若i是j的亲戚则i的所有亲戚都是j的亲戚。有P次询问某两人是否具有亲戚关系。
分析:具有亲戚关系的人都在同一个集合里,符合并查集的性质。本题数据范围不大,N<=5000,可以不用优化。
代码:
# include <cstdio>
const int SIZE=5000+5;
int n , m , p , f[SIZE] ;
int _find ( int ) ; //求集合的根
void _union ( int , int ) ; //合并两个集合
int main () {
scanf ( "%d%d%d" , &n , &m , &p ) ;
for ( int i=0 ; i!=n+1 ; ++i ) f[i]=i ; //初始化
for ( int i=0 ; i!=m ; ++i ) {
int A , B ;
scanf ( "%d%d" , &A , &B ) ;
_union ( A , B ) ;
}
for ( int i=0 ; i!=p ; ++i ) {
int A , B ;
scanf ( "%d%d" , &A , &B ) ;
if ( _find(A)==_find(B) ) printf ( "Yes\n" ) ; else printf ( "No\n" ) ;
}
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 ;
}