算法-并查集-臭虫也疯狂

题目内容：

 霍普教授研究臭虫的性取向。实验前他弄清楚了n个臭虫的性别，并在臭虫的背上标了数字编号（1~n）。现在给一批臭虫的编号配对，要检查里面有没有同性恋。

输入描述：

第一行是整数c,下面接着有c个测试用例。

每个测试用例的第一行是臭虫的数目n（1~2000），以及配对的数目m（1~10^6）。接下来的行就是m个配对的臭虫编号.

输出描述：

一共c行， 每行打印“testcase i:没有发现同性恋”,或者“testcase i:发现同性恋”

输入样例：

2

3 3

1 2

2 3

1 3

4 2

1 2

3 4

输出样例：

testcase 1:发现同性恋

testcase 2:没有发现同性恋

代码： 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <time.h> 
using namespace std;
int n, m;
const int MAX = 2005;
int pre[MAX];
int c[MAX];
int find(int x){
	if(pre[x] == x){
		return x;
	}
	return pre[x] = find(pre[x]);
}
 
void unit(int x, int y){
	pre[find(x)] = find(y);
}
 
int init(){
	for(int i = 1; i <= n; i++){
		pre[i] = i;
	}
	memset(c, 0, sizeof(c));
	return 0;
}
 
int main(){
 	int t;
 	clock_t startTime,endTime;
 	scanf("%d", &t);
 	startTime = clock();
 	for(int k = 1; k <= t; k++){
		scanf("%d %d", &n, &m);
 		init();
 		int flag = 1;
 		for(int i = 0; i < m; i++){
 			int a, b;
 			scanf("%d %d", &a, &b);
 			if(c[a] == 0 && c[b] == 0){
 				c[a] = b;
 				c[b] = a;
 			}
 			else if(c[a] && c[b] == 0){
 				unit(c[a], b);
 				c[b] = a;
 			}
 			else if(c[a] == 0 && c[b]){
 				unit(a, c[b]);
 				c[a] = b;
 			}
 			else{
 				unit(a, c[b]);
 				unit(b, c[a]);
 			}
 			if(find(a) == find(b)){
 				flag = 0;
 			}
 		}
 		if(flag == 0){
 			printf("testcase %d:发现同性恋\n", k);
		}
		else{
			printf("testcase %d:没有发现同性恋\n", k);
		}	
 	}
 	endTime = clock();
 	printf("The run time is: %d ms",(endTime - startTime));
	return 0;
}