POJ 虫子的生活 解题报告 （种类并查集）

POJ 虫子的生活 解题报告

题目链接: http://dsalgo.openjudge.cn/tree/10/

思路：

这种涉及到多个类别（在这里的意思是产生过相互关系的虫子），和少量集合（这里指性别）的题目，我想到的是使用并查集，但是基础的并查集并不能解决这种问题，因为这里有两种类（类别和集合），应该使用升级版的种类并查集。

种类并查集的两个要点是：

1. 处理集合

2. 更新集合

我把实现细节放在了代码里：

#include <cstdio>

int Set[2010]; //并查集的父节点
int relaWithFarther[2010]; //记录与父节点的关系 0 same Sex, 1 diff Sex
bool suspicious;

void ini()
{
suspicious = false;
for(int i = 0; i < 2010; ++i)
{
Set[i] = i;
relaWithFarther[i] = 0;
}
}

int find(int node)
{
if(Set[node] == node)
return node;
int end = find(Set[node]);
relaWithFarther[node] = relaWithFarther[node] ^ relaWithFarther[Set[node]];
//到这里node的父节点的父节点已经压缩到了end，node与end的关系可以由node与其现在的父节点的关系以及Set[node]与其父节点的关系推出来。
Set[node] = end;
return end;
}

void Union(int nodeA, int nodeB)
{
int endA = find(nodeA);
int endB = find(nodeB);
//注意因为路径压缩，nodeA和nodeB的父节点已经转变。
if(endA == endB)
{
if(relaWithFarther[nodeA] == relaWithFarther[nodeB])//同性
{
suspicious = true;
return ;
}
}
//此时保证了非同性
Set[endA] = endB;
relaWithFarther[endA] = relaWithFarther[nodeA] ^ relaWithFarther[nodeB] ^ 1;
//同理推出更新规则
}

int main()
{
int cases;
int m, n;
int nodeA, nodeB;
scanf("%d", &cases);
for(int t = 1; t <= cases; ++t) {
ini();
scanf("%d %d", &m, &n);
for(int i = 0; i < n; ++i){
scanf("%d %d", &nodeA, &nodeB);
Union(nodeA, nodeB);
}
printf("Scenario #%d:\n", t);
if(suspicious){
printf("Suspicious bugs found!\n\n");
}else{
printf("No suspicious bugs found!\n\n");
}
}
return 0;
}