【带权并查集：区间统计】-- poj1733

题目来源：poj 1733

参考：kungbin大神的解题报告

思路：

• 假如输入的[left,right]判断是同一个集合的，则奇偶性关系已经知道了。
  例子：
  
• 加入输入的[left,right]不是同一个集合的，则需要unite。
  例子：
  如果已知[5,6]的奇偶性为偶数，[9,12]的奇偶性为奇数，新的输入为[7,12,odd]。
  则sum[6] = 1 ^ sum[8]^ sum[4]。

用map写的超时了。。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <map>
using namespace std;
const int maxn = 15000;
int n, m;
int father[maxn];
int sum[maxn];    //从当前结点到根节点的1的个数的奇偶性(0为偶数，1为奇数)
map<int, int> mp; //把输入数据重新编号为0到n-1
void init()
{
    for (int i = 0; i <= n; i++)
    {
        father[i] = i;
        sum[i] = 0;
    }
}

int find(int x)
{
    if (father[x] == x)
    {
        return x;
    }
    else
    {
        int tf = father[x];
        father[x] = find(father[x]);
        sum[x] ^= sum[tf]; //与父结点到根节点的奇偶性做异或操作，就是当前结点到根节点的奇偶性
        return father[x];
    }
}

bool unite(int x, int y, int value)
{
    int faX = find(x);
    int faY = find(y);
    if (faX != faY)
    {
        father[faX] = faY;
        sum[faX] = sum[y] ^ value ^ sum[x];
    }
    else
    {
        if (sum[x] ^ sum[y] != value)
        {
            //若x和y奇偶性相同，则[x,y]应该是偶数,则异或结果为0，判断value此时是否为0
            return false;
        }
    }
    return true;
}

int main()
{
    while (scanf("%d", &n) == 1)
    {
        init();
        mp.clear();
        cin >> m;
        int u, v;
        int index = 0;
        int ans = m;
        for (int i = 1; i <= m; i++)
        {
            char str[10];
            scanf("%d%d%s", &u, &v, str);
            if (u > v)
            {
                swap(u, v);
            }
            if (ans < m)
            {
                continue;
            }
            u--;
            if (mp.find(u) == mp.end())
            {
                mp[u] = index++;
            }
            if (mp.find(v) == mp.end())
            {
                mp[v] = index++;
            }
            if (str[0] == 'e') //偶
            {
                if (!unite(mp[u], mp[v], 0))
                {
                    ans = i - 1;
                }
            }
            else if (str[0] == 'o') //奇
            {
                if (!unite(mp[u], mp[v], 1))
                {
                    ans = i - 1;
                }
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}