感冒病毒 suspects 并查集

题目大意:给定n个人,m个团,同一个团中只要有一个人感染病毒整个团就都感染病毒。现在0号感染了病毒,问总共有多少个人会感染病毒。


题目分析:简单的并查集。

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,n) for(int i=0;i=0;i--)
#define MEM(a) memset(a,0,sizeof(a))
#define MEMI(a) memset(a,127,sizeof(a))
#define MEMi(a) memset(a,128,sizeof(a))
#define INF (2139062143)
#define phiF (1000000006)
#define MAXN (1000000+10)
typedef long long LL;


int fa[30005],n,m,k,x,y,fx,fy;

int get(int v){
	if (fa[v]!=v) fa[v]=get(fa[v]);
	return fa[v];
}

int main(){
	scanf("%d%d",&n,&m);
	Fork (i,0,n) fa[i]=i;
	
	For (i,m) {
		scanf("%d",&k);
		if (k){
			scanf("%d",&x);
			For (j,k-1){
				scanf("%d",&y);
				int fx,fy;
				fx=get(x);
				fy=get(y);
				if (fx!=fy) fa[fx]=fy;
			}
		}
	}
	fx=get(0);int ans=1;
	For (i,n) {
		fy=get(i);
		if (fx==fy) ans++;
	}
	printf("%d",ans);
}

你可能感兴趣的:(并查集,并查集)