POJ 1733 Parity game【带权并查集+离散化处理】

POJ 1733 Parity game

题目大意:
按顺序给定若干描述:区间l到r中1的个数是奇数个(偶数个)。
问:从哪一条开始,下一条与前面的相违,错误了。若全是对的,则输出描述条数。

具体思路:
这题与HDU 3080很相似,思路是一样的,只是路径压缩的时对权值的处理不一样,这里需要用异或来实现。
附上该题题解:How many answers are wrong
不同的是这里的n是1e9,无法开出那么大的数组,因此需要离散化处理(第一次接触),可用map实现。

具体代码:

#include
#include
#include
#include
#include
using namespace std;

const int N = 1e4 + 5;
int fa[N];
int val[N];
map<int, int>mp;
int total;
void init()
{
     
	mp.clear();
	total = 0;
	memset(fa, -1, sizeof(fa));
	memset(val, 0, sizeof(val));
}
int insert(int x)	//map实现数据离散化
{
     
	//x是键,total是值,每个离散的键映射成一个值
	if (mp.find(x) == mp.end())mp[x] = total++;
	return mp[x];
}
int find(int son)
{
     
	if (-1 == fa[son])
		return son;
	int tmp = find(fa[son]);
	val[son] ^= val[fa[son]];
	return fa[son] = tmp;
}
bool unite(int u, int v, int w)
{
     
	int r1 = find(u);
	int r2 = find(v);
	if (r1 == r2) {
     
		if (val[u] ^ val[v] == w)
			return true;
		return false;
	}
	else {
     	//不在同一集合
		fa[r1] = r2;	//合并集合
		val[r1] = val[v] ^ val[u] ^ w;	//更新val数组
		return true;
	}
}

int main()
{
     
	int n, m;
	int u, v;
	char str[20];
	while (~scanf("%d%d", &n, &m))
	{
     
		init();
		int ans = m;
		for (int i = 0; i < m; i++)
		{
     
			scanf("%d%d%s", &u, &v, &str);
			if (u > v)swap(u, v);
			if (ans < m)continue;	//找到一个错误的,就不再判断,只是输入
			u = insert(u - 1);
			v = insert(v);
			int w;
			if (str[0] == 'e')w = 0;
			else w = 1;
			if (!unite(u, v, w))ans = i;
		}
		printf("%d\n", ans);
	}
	return 0;
}

你可能感兴趣的:(并查集,OJ题解)