臭蟲也瘋狂
霍普教授研究臭蟲的性取向。實驗前他弄清楚了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;
}