无间道之并查集 （hihocoder第十四周i）

时间限制:20000ms

单点时限:1000ms

内存限制:256MB

描述

这天天气晴朗、阳光明媚、鸟语花香，空气中弥漫着春天的气息……额，说远了，总之，小Hi和小Ho决定趁着这朗朗春光出去玩。

但是刚刚离开居住的宾馆不久，抄近道不小心走入了一条偏僻小道的小Hi和小Ho就发现自己的前方走来了几个彪形大汉，定睛一看还都是地地道道的黑人兄弟！小Hi和小Ho这下就慌了神，捡肥皂事小，这一身百把来斤别一不小心葬身他乡可就没处说去了。

就在两人正举足无措之时，为首的黑叔叔从怀里掏出了一件东西——两张花花绿绿的纸，分别递给了小Hi和小Ho。

小Hi和小Ho接过来，只见上面写道（译为中文）：“本地最大的帮派——青龙帮，诚邀您的加入！”下面还详细的列出了加入青龙帮的种种好处。

于是两人略感心安，在同黑叔叔们交谈一番之后，已是均感相见恨晚。同时，在小Hi和小Ho表示自己不日便将回国之后，黑叔叔们也没有再提加入帮派之事，但是那为首的黑叔叔思索一会，开口道（译为中文）：“我现在有一个难题，思索了很久也没法子解决，既然你们俩都是高材生，不如来帮我看看。”

小Hi和小Ho点了点头表示没问题，于是黑叔叔继续说道：“这个问题是这样的，我们帮派最近混进了许多警察的卧底，但是在我们的调查过程中只能够知道诸如‘某人和另一个人是同阵营的’这样的信息，虽然没有办法知道他们具体是哪个阵营的，但是这样的信息也是很重要的，因为我们经常会想要知道某两个人究竟是不是同一阵营的。”

小Hi和小Ho赞同的点了点头，毕竟无间道也都是他们看过的。

黑叔叔接着说道：“于是现在问题就来了，我希望你们能写出这样一个程序，我会有两种操作，一种是告诉它哪两个人是同一阵营的，而另一种是询问某两个人是不是同一阵营的……既然你们就要回国了，不如现在就去我们帮派的总部写好这个程序再走把。”

为了生命安全与……小Hi和小Ho都不得不解决这个问题！那么他们究竟从何下手呢？

提示：说起来其实就是不断的合并集合嘛~

输入

每个测试点（输入文件）有且仅有一组测试数据。

每组测试数据的第1行为一个整数N，表示黑叔叔总共进行的操作次数。

每组测试数据的第2~N+1行，每行分别描述黑叔叔的一次操作，其中第i+1行为一个整数op_i和两个由大小写字母组成的字符串Name1_i, Name2_i，其中op_i只可能为0或1，当op_i=0时，表示黑叔叔判定Name1_i和Name2_i是同一阵营的，当op_i=1时，表示黑叔叔希望知道Name1_i和Name2_i是否为同一阵营的。

对于100%的数据，满足N<=10^5, 且数据中所有涉及的人物中不存在两个名字相同的人（即姓名唯一的确定了一个人），对于所有的i，满足Name1_i和Name2_i是不同的两个人。

输出

对于每组测试数据，对于黑叔叔每次op_i=1的操作，输出一行，表示查询的结果：如果根据已知信息（即这次操作之前的所有op_i=0的操作），可以判定询问中的两个人是同一阵营的，则输出yes，否则输出no。

样例输入

10

0 Steven David

0 Lcch Dzx

1 Lcch Dzx

1 David Dzx

0 Lcch David

0 Frank Dzx

1 Steven Dzx

1 Frank David

0 Steven Dzx

0 Dzx Frank

样例输出

yes

no

yes

yes

并查集简单题


朴素版本（路径压缩）

 1 #include "iostream"

 2 #include "algorithm"

 3 #include "map"

 4 

 5 using namespace std;

 6 

 7 map< string, int > s;

 8 int father[111111];

 9 

10 int find_set(int x)

11 {

12     return father[x] = (x == father[x]) ? x : find_set(father[x]);

13 }

14 

15 void Union(int a,int b)

16 {

17     int x = find_set(a);

18     int y = find_set(b);

19     if (x != y) father[x] = y;

20 }

21 

22 int main()

23 {

24     int n, c = 1;

25     bool ok;

26     cin >> n;

27     for (int i = 0;i <= n; ++ i)

28         father[i] = i;

29     while (n--) {

30         string a,b;

31         cin >> ok >> a >> b;

32         if (s[a] == 0)  s[a] = c++;

33         if (s[b] == 0)  s[b] = c++;

34         if (ok) {

35             if (find_set(s[a]) == find_set(s[b]) )

36                 cout << "yes" << endl;

37             else cout << "no" << endl;

38         }

39         else {

40             Union(s[a],s[b]);

41         }

42     }

43     return 0;

44 }

代码君

 

路径压缩 使树平衡

 1 #include "iostream"

 2 #include "algorithm"

 3 #include "map"

 4 #include "string"

 5 #include "memory.h"

 6 using namespace std;

 7 

 8 map< string, int > s;

 9 int father[111111];

10 

11 int find_set(int x)

12 {

13     if(father[x]<0) return x;

14     return father[x] = find_set(father[x]);

15 }

16 

17 void Union(int a,int b)

18 {

19     int x = find_set(a);

20     int y = find_set(b);

21     if (x > y) {

22         father[y] += father[x];

23         father[x] = y;

24     }

25     if (x < y)  {

26         father[x] += father[y];

27         father[y] = x;

28     }

29 }

30 

31 int main()

32 {

33     int n, c = 1;

34     bool ok;

35     cin >> n;

36     memset(father,-1,sizeof(father));

37     while (n--) {

38         string a,b;

39         cin >> ok >> a >> b;

40         if (s[a] == 0)  s[a] = c++;

41         if (s[b] == 0)  s[b] = c++;

42         if (ok) {

43             if ( find_set(s[a]) == find_set(s[b]) )

44                 cout << "yes" << endl;

45             else cout << "no" << endl;

46         }

47         else {

48             Union(s[a],s[b]);

49         }

50     }

51     return 0;

52 }

代码君

 

按秩压缩

 1 #include "iostream"

 2 #include "algorithm"

 3 #include "map"

 4 #include "string"

 5 #include "memory.h"

 6 using namespace std;

 7 

 8 map< string, int > s;

 9 int father[111111];

10 int rank[111111];

11 int find_set(int x)

12 {

13     if(father[x] == x) return x;

14     return father[x] = find_set(father[x]);

15 }

16 

17 void Union(int a,int b)

18 {

19     int x = find_set(a);

20     int y = find_set(b);

21     if (rank[x] <= rank[y]) {

22         father[x] = y;

23         if (rank[x] == rank[y])

24             rank[y]++;

25     } else {

26       father[y] = x;

27     }

28 }

29 

30 int main()

31 {

32     int n, c = 1;

33     bool ok;

34     cin >> n;

35     for (int i = 0;i < 100000; ++ i)

36         father[i] = i;

37     while (n--) {

38         string a,b;

39         cin >> ok >> a >> b;

40         if (s[a] == 0)  s[a] = c++;

41         if (s[b] == 0)  s[b] = c++;

42         if (ok) {

43             if ( find_set(s[a]) == find_set(s[b]) )

44                 cout << "yes" << endl;

45             else cout << "no" << endl;

46         }

47         else {

48             Union(s[a],s[b]);

49         }

50     }

51     return 0;

52 }

代码君