欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

93-臭虫也疯狂

程序员文章站 2022-05-22 11:33:51
...

                   臭虫也疯狂
霍普教授研究臭虫的性取向。实验前他弄清楚了n个臭虫的性别,并在臭虫的背上标了数
字编号(1~n)。现在给一批臭虫的编号配对,要检查里面有没有同性恋。
输入描述
第一行是整数c,下面接着有c个测试用例。
每个测试用例的第一行是臭虫的数目n(12000),以及配对的数目m(110^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 <cstring>
using namespace std;
int n, m;
const int MAX = 100005;
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));
}
 
int main(){
 	int t;
 	cin >> t;
 	for(int k = 1; k <= t; k++){
// 		cin >> n >> m;
 		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;
// 				break;
 			}
 		}
 		if(flag == 0){
 			printf("testcase %d:发现同性恋\n", t);
		}
		else{
			printf("testcase %d:没有发现同性恋\n", t);
		}
 	}
 
	return 0;
}