Description
现代社会通信便捷,借助于Internet形成了各式各样的社区,每个人都可能属于多个社交圈,尤其是Facebook类社交网站的出现,使世界缩小了,人与人的交往扩大了频繁了。sed同学正在做这方面的毕业设计课题,指导老师给他布置了一个任务:已知一群人的社会关系网络,判断两个人之间的关系,他们是否可以通过社交圈的人相互结识。
Input
第一行包括三个整数:n、 m、k,分别表示人数、社区数、查询两个人之间的关系的用例数 (1 ≤ n ≤ 10000, 0 ≤ m ≤ 100,1 ≤ k ≤ 100)。
m行,每行首先给出一个社区的人数,然后给出代表人的序号。
k行,每行给出待查询的两个人(用序号表示)。
Output
输出k行,每行给出两个人(用序号表示)、YES或NO, YES表示这两个人可以通过社交圈的人相互结识,NO表示不能。
注意:输出部分的结尾要求包含一个多余的空行。
Sample Input
3 1 2
2 1 2
0 1
1 2
Sample Output
0 1 NO
1 2 YES
学校比赛时的一道题目,当时没有做出来。回来后猛然发现是并查集。。。可是这题的数据结构的实现较简单,根本不需要用到二维数组或者链表来保存图,只需要一个一维数组足以:
一个集合的代表元素在数组中以负数的形式保存此集合的元素个数:比如一个集合的元素个数为5,则在相应的数组位置保存-5;
一个集合中的其他元素在数组中保存代表元素在数组中的位置:比如一个集合6, 4,3, 8,1。则在数组中依次保存的是-5,6, 6,6, 6;
如果两个集合合并,则只要改动一个集合中元素在数组中的值,和合并后代表元素的值。
以下代码在两个集合合并后没有改变代表元素的值,不过任然AC,因为题目不需要这个值:)
![](https://www.cnblogs.com/Images/OutliningIndicators/ContractedBlock.gif)
Code
1 #include <stdio.h>
2 int n, m, k;
3 int G[10001];
4 int main()
5 {
6 int i, gpn, j, head, t, a, b, flag;
7 int p[10001];
8 //freopen("input.txt", "r", stdin);
9 scanf("%d %d %d", &n, &m, &k);
10 for (i=0;i<m;i++){
11 scanf("%d", &gpn);
12 for (j=1;j<=gpn;j++){
13 scanf("%d", p+j);
14 }
15 for (j=1;j<=gpn;j++){
16 if (G[p[j]]==0){
17 if (j==1){
18 G[p[j]]=-gpn;
19 head=p[j];
20 }
21 else{
22 G[p[j]]=head;
23 }
24 }
25 else{
26 head=G[p[j]];
27 for (t=1;t<=gpn;t++){
28 G[p[t]]=head;
29 }
30 break;
31 }
32 }
33 }
34
35 for (i=0;i<k;i++){
36 scanf("%d %d", &a, &b);
37 if (G[a]!=0 && G[b]!=0){
38 if (G[a]==G[b]){
39 flag=1;
40 }
41 else if (a==G[b]){
42 flag=1;
43 }
44 else if (b==G[a]){
45 flag=1;
46 }
47 else{
48 flag=0;
49 }
50 }
51 else{
52 flag=0;
53 }
54 if (flag==1){
55 printf("%d %d YES\n", a, b);
56 }
57 else{
58 printf("%d %d NO\n", a, b);
59 }
60 }
61 return 0;
62 } END