1. 程式人生 > >L2-024. 部落(並查集應用)

L2-024. 部落(並查集應用)

在一個社群裡,每個人都有自己的小圈子,還可能同時屬於很多不同的朋友圈。我們認為朋友的朋友都算在一個部落裡,於是要請你統計一下,在一個給定社群中,到底有多少個互不相交的部落?並且檢查任意兩個人是否屬於同一個部落。

輸入格式:

輸入在第一行給出一個正整數N(<= 104),是已知小圈子的個數。隨後N行,每行按下列格式給出一個小圈子裡的人:

K P[1] P[2] ... P[K]

其中K是小圈子裡的人數,P[i](i=1, .., K)是小圈子裡每個人的編號。這裡所有人的編號從1開始連續編號,最大編號不會超過104

之後一行給出一個非負整數Q(<= 104),是查詢次數。隨後Q行,每行給出一對被查詢的人的編號。

輸出格式:

首先在一行中輸出這個社群的總人數、以及互不相交的部落的個數。隨後對每一次查詢,如果他們屬於同一個部落,則在一行中輸出“Y”,否則輸出“N”。

輸入樣例:
4
3 10 1 2
2 3 4
4 1 5 7 8
3 9 6 4
2
10 5
3 7
輸出樣例:
10 2
Y
N

簡單的並查集應用題,閒的時候水了水。對並查集兩個子函式並不能掌握很好。

題意:給出很多個集合,求總人數、集合個數,給任意兩個人的編號,求兩人是否在同一集合。

程式碼如下:

#include<cstdio>
int pre[10010];

int find(int x){
	int r=x;
	while(pre[r]!=r)
		r=pre[r];
	int i=x,temp;
	while(i!=r){
		temp=pre[i];
		pre[i]=r;
		i=temp;
	}
	return r;
}

void join(int x,int y){
	int xx=find(x);
	int yy=find(y);
	if(xx!=yy)
		pre[xx]=yy;
}

int main(){
	for(int i=0;i<10010;i++)
		pre[i]=i;
	int n,max=-1;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		int k;
		scanf("%d",&k);
		int tt;
		scanf("%d",&tt);
		if(tt>max)
				max=tt;
		for(int j=1;j<k;j++){
			int t;
			scanf("%d",&t);
			if(t>max)
				max=t;
			join(t,tt);
		}
	}
	int sum=0;
	for(int i=1;i<=max;i++)
		if(pre[i]==i)
			sum++; 
	printf("%d %d\n",max,sum);
	int q;
	scanf("%d",&q);
	for(int i=0;i<q;i++){
		int x,y;
		scanf("%d%d",&x,&y);
		if(find(x)==find(y))
			printf("Y\n");
		else
			printf("N\n");
	}
	return 0;
}