问题描述:
假设N个人参加宴会(标号从0到N-1),1和2交流后,1就同2建立了认识关系,2和3交流后,2与3就建立了认识关系,由于传递性1和3也具备认识关系。由于人们之间的兴趣不同,经过一场宴会后,N个人可能形成若干个小群体。给定参加宴会的人数以及宴会上人员之间彼此交流的情况,求宴会后形成的小群体的数目。
限制:
空间复杂度为O(n)
输入从文件中读取,文件名为sample.txt
例如输入:
8
1 2
4 5
3 6
2 7
1 7
3 5
4 6
2 1
输出:
3
由于限制空间复杂度为O(n),所以我们只能使用一维数组。
若两个元素位于同一个组,则为这两个元素建立树的关系。
设想有一个一维数组,数组的长度即为参加宴会的人的个数,标号0~n-1的每个人与数组中的元素下标一一对应。初始时,将数组中的所有元素的值都设为-1,代表每个元素都是根节点,说明此时有n个组。若标号为a与标号为b的人交流,则分别找到标号为a和标号为b所属的组的根元素所对应的标号,如何判断是否是根元素呢?若元素的值为-1,则代表是根元素。那么如何找到某个标号a所属的组的根元素呢?先看这个标号对应的元素的值是否为-1,若为,则它就是根元素,则a就是根元素的标号;若不为-1,则它的值就是父元素的标号,于是我们继续查找父元素是否为根元素,依此类推,直到找到根元素。
当我们找到标号为a和标号为b所属的组的根元素所对应的标号后(假设分别为m,n),判断这两个标号是否相同,若相同,就继续下一条;若不相同,说明这两个组属于同一个组,则必须将这两个组合并,如何合并呢?只需让一个组的根元素作为另外一个组的根元素的孩子,即将下标为n的元素的值修改为m,这样下标为m的元素即变为了下标为n的元素的父节点。
以下代码是查找标号为idx所属组的根元素所对应的标号
/*
* 查找每个组的根,-1代表根
*/
int find_root(int * arr, int idx)
{
while (arr[idx] != -1)
{
idx = arr[idx];
}
return idx;
}
以下是程序的主体:
#include <stdio.h>
#include <stdlib.h>
int main()
{
int count;
FILE * infile = fopen("sample.txt", "r");
fscanf(infile, "%d", &count);
int * arr = (int *) malloc(count * sizeof(int));
for (int i = 0; i < count; ++i)
arr[i] = -1;
int a, b;
while (fscanf(infile, "%d %d", &a, &b) != EOF)
{
int root_a = find_root(arr, a);
int root_b = find_root(arr, b);
if (root_a != root_b) //说明不是同一组
{
arr[root_b] = root_a; // 将以root_b为根的组合并到以root_a为根的组
}
}
// 此时,数组中-1的个数,即为组的个数
int group_count = 0;
for (int i = 0; i < count; ++i)
{
if (arr[i] == -1)
{
group_count++;
}
}
printf("%d", group_count);
free(arr);
fclose(infile);
return 0;
}
最后数组中-1的个数即为组的个数。