POJ 1733 Parity game(离散化+带权并查集)

题目链接：Parity game

题意：有一个全是0，1的序列，给定一些区间[l,r]并且区间1的个数是奇数偶数，求最早矛盾位置

思路：带权并查集，对于区间l, r，如果0, l和0, r相同，那么[l,r]为偶数，如果不同为奇数，那么每次执行并查集只要带权，rank = 0表示和祖先相同，rank = 1表示和祖先不同，注意发现矛盾就呀break；不然会wa。坑点。。

代码:

#include <stdio.h>
#include <string.h>
#include <map>
using namespace std;

const int N = 5005;
int n, m, parent[N], Max, rank[N];
map<int, int> vis;

int find(int x) {
    if (x == parent[x])
	return x;
    int px = find(parent[x]);
    rank[x] ^= rank[parent[x]];
    return parent[x] = px;
}

void init() {
    vis.clear();
    Max = 0;
    for (int i = 0; i < N; i++) {
	parent[i] = i;
	rank[i] = 0;
    }
}

int main() {
	scanf("%d%d", &n, &m);
	init();
	int a, b, flag = 0;
       	char q[10];
	int i;
	for (i = 0; i < m; i++) {
	    scanf("%d%d%s", &a, &b, q);
	    int d = 0;
	    if (q[0] == 'o') d = 1;
	    a--;
	    if (!vis.count(a)) {
		vis[a] = Max;
		Max++;
	    }
	    if (!vis.count(b)) {
		vis[b] = Max;
		Max++;
	    }
	    int aa = vis[a], bb = vis[b];
	    a = aa; b = bb;
	    int pa = find(a), pb = find(b);
	    if (pa != pb) {
		parent[pa] = pb;
		rank[pa] = rank[a]^rank[b]^d;
	    }
	    else {
		if ((rank[a]^rank[b]) != d)
		    break;
	    }
	}
	printf("%d\n", i);
    return 0;
}