A Bug's Life，并查集

题目内容：臭虫也疯狂

霍普教授研究臭虫的性取向。实验前他弄清楚了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:没有发现同性恋
可坑了我好久的时间， 并查集的简单应用应该是，但还是绕进去了， 并查集应使得同性之间成为一个群体， 故应多开一个数组存储其对象， 最后还不能用cin， cout， 改为c的才堪堪过了校oj

#include<iostream> 
#include<cstdio>
#include<vector>
#include<cstring>	
#define lowbit(x) ((x)&(-x))
#define maxn 100005
using namespace std;

int pre[100000];
int l[100000];//用于存储对象
int t, n, m;
void init(int a)
{
	for (int i = 1; i <= a; i++)
		pre[i] = i;
}

int find(int a)
{
	if (pre[a] == a)
		return a;
	return pre[a] = find(pre[a]);
}

void unit(int a, int b)
{
	if (find(a) != find(b))
		pre[a] = b;
}

int main()
{
	scanf("%d", &t);
	int k = 0;
	while (t--)
	{
		scanf("%d %d", &n, &m);
		init(n);
		int a, b;
		bool flag = false;
		for (int i = 0; i < m; i++)
		{
			scanf("%d %d", &a, &b);
			if (flag) continue;
			if (!l[a] && !l[b])
			{
				l[a] = b;
				l[b] = a;
			}
			else if (l[a] && !l[b]) l[b] = a, unit(b, l[a]);//如果两个臭虫原来都还没有对象，那么他们两个互为异性
			else if (!l[a] && l[b]) l[a] = b, unit(a, l[b]);//如果其中一个有对象，那么这个新的对象和原来那个是同性的，需要进行并查集的合并
			else unit(l[a], b), unit(a, l[b]);
			if (find(a) == find(b)) flag = true;
		}
		printf("testcase %d:", ++k);
		if (flag)
			printf("发现同性恋\n");
		else
			printf("没有发现同性恋\n");
	}
}