本文實例講述了C++并查集親戚(Relations)算法。分享給大家供大家參考。具體分析如下:
題目: 親戚(Relations)
或許你并不知道,你的某個朋友是你的親戚。他可能是你的曾祖父的外公的女婿的外甥的表姐的孫子。如果能得到完整的家譜,判斷兩個人是否親戚應該是可行的,但如果兩個人的最近公共祖先與他們相隔好幾代,使得家譜十分龐大,那么檢驗親戚關系實非人力所能及.在這種情況下,最好的幫手就是計算機。
為了將問題簡化,你將得到一些親戚關系的信息,如同Marry和Tom是親戚,Tom和B en是親戚,等等。從這些信息中,你可以推出Marry和Ben是親戚。請寫一個程序,對于我們的關心的親戚關系的提問,以最快的速度給出答案。
參考輸入輸出格式 輸入由兩部分組成。
第一部分以N,M開始。N為問題涉及的人的個數(1 ≤ N ≤ 20000)。這些人的編號為1,2,3,…,N。下面有M行(1 ≤ M ≤ 1000000),每行有兩個數ai, bi,表示已知ai和bi是親戚.
第二部分以Q開始。以下Q行有Q個詢問(1 ≤ Q ≤ 1 000 000),每行為ci, di,表示詢問ci和di是否為親戚。
對于每個詢問ci, di,若ci和di為親戚,則輸出Yes,否則輸出No。
樣例輸入與輸出
輸入
10 7
2 4
5 7
1 3
8 9
1 2
5 6
2 3
3
3 4
7 10
8 9
輸出
Yes
No
Yes
如果這道題目不用并查集,而只用鏈表或數組來存儲集合,那么效率很低,肯定超時。
代碼如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#include <iostream> #include <cstdio> using namespace std; int father[20010]; //father[i]表示i的父親 int Find( int a) //查找其父親并壓縮路徑 { if (father[a] != a) father[a] = Find(father[a]); return father[a]; } int main() { int N,M; int a,b; scanf ( "%d%d" ,&N,&M); //給每個元素建立一個集合 for ( int i = 1 ; i <= N ; ++i) father[i] = i; //合并 for ( int i = 0 ; i < M ; ++i) { scanf ( "%d%d" ,&a,&b); a = Find(a); b = Find(b); father[a] = b; } //查詢 scanf ( "%d" ,&M); while (M--) { scanf ( "%d%d" ,&a,&b); a = Find(a); b = Find(b); if (a == b) printf ( "YES\n" ); else printf ( "NO\n" ); } return 0; } |
希望本文所述對大家的C++程序設計有所幫助。