A Bug's Life POJ - 2492 带权并查集

题解

题目大意，给你一些情侣问其中是否有同性恋，输入的数据表示是一对情侣的编号

使用带权并查集求解，与题目食物链类似，可以说是食物链的弱化版，定义与父节点关系时用0表示同性1表示异性
同性+同性还是同性，异性+异性是同性，同性+异性是异性

AC代码

#include <stdio.h>
#include <iostream>
#define fst first
#define sed second
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const int MAXN = 2e6 + 10;
int f[MAXN], r[MAXN]; //父节点 与父节点是同性0还是异性1

int find(int x)
{
	if (f[x] != x)
	{
		int fx = f[x];
		f[x] = find(f[x]);
		r[x] = (r[x] + r[fx]) % 2; //同性的同性是同性 异性的异性是同性 同性的异性是异性
	}
	return f[x];
}
void join(int x, int y, int z)
{
	int fx = find(x), fy = find(y);
	if (fx != fy)
	{
		f[fx] = fy;
		r[fx] = (r[x] + z + r[y]) % 2; //关系箭头反转01不变 fx->x->y->fy
	}
}
int main()
{
#ifdef LOCAL
	freopen("C:/input.txt", "r", stdin);
#endif
	int T;
	cin >> T;
	for (int ti = 0; ti < T; ti++)
	{
		int n, m;
		scanf("%d%d", &n, &m);
		for (int i = 1; i <= n; i++)
			f[i] = i, r[i] = 0; //自己和自己是同性
		int flag = 0;
		while (m--)
		{
			int x, y;
			scanf("%d%d", &x, &y);
			if (find(x) == find(y) && r[x] == r[y]) //可以确定关系并且和父节点关系一致则说明xy同性
				flag = 1;
			else
				join(x, y, 1);
		}
		if (ti)
			putchar('\n');
		printf("Scenario #%d:\n", ti + 1);
		if (flag)
			printf("Suspicious bugs found!\n");
		else
			printf("No suspicious bugs found!\n");
	}

	return 0;
}