L2-024. 部落(並查集應用)
阿新 • • 發佈:2019-02-18
在一個社群裡,每個人都有自己的小圈子,還可能同時屬於很多不同的朋友圈。我們認為朋友的朋友都算在一個部落裡,於是要請你統計一下,在一個給定社群中,到底有多少個互不相交的部落?並且檢查任意兩個人是否屬於同一個部落。
輸入格式:
輸入在第一行給出一個正整數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; }