poj 1733(带权并查集)

题意:有n个结点,给出了q个操作,操作是a b string表示结点a到结点b的和是奇数或偶数,输出x(前x个操作都是正确的)。
题解:带权并查集经典题,因为结点可能有10,000,000,000个,所以需要离散化,不用的点就不考虑了。因为要a加到b,如果a==b无法找到各自的根节点并判断是否要合并,所以其实是mp[a - 1]到mp[b]并入集合。另外需要注意结点是有顺序的,父亲结点要大于等于子节点,这样判断才不会出错,所以合并时要考虑两个根结点的大小有不同关系式。

#include <stdio.h>
#include <map>
using namespace std;
const int N = 10005;
int pa[N], n, q, cnt, rel[N], l[N], r[N];
char str[N][10];
map<int, int> mp;

int get_parent(int x) {
    if (pa[x] != x) {
        int px = get_parent(pa[x]);
        rel[x] = (rel[x] + rel[pa[x]]) % 2;
        pa[x] = px;
    }
    return pa[x];
}

int main() {
    while (scanf("%d", &n) == 1) {
        scanf("%d", &q);
        mp.clear();
        cnt = 1;
        for (int i = 1; i <= q; i++) {
            scanf("%d%d%s", &l[i], &r[i], str[i]);
            if (!mp[l[i] - 1])
                mp[l[i] - 1] = cnt++;
            if (!mp[r[i]])
                mp[r[i]] = cnt++;
        }
        for (int i = 0; i <= cnt; i++) {
            pa[i] = i;
            rel[i] = 0;
        }
        int res = -1;
        for (int i = 1; i <= q; i++) {
            int flag = str[i][0] == 'e' ? 0 : 1;
            int px = get_parent(mp[l[i] - 1]);
            int py = get_parent(mp[r[i]]);
            if (px != py) {
                if (px < py) {
                    pa[px] = py;
                    rel[px] = (flag + rel[mp[r[i]]] - rel[mp[l[i] - 1]] + 2) % 2;
                }
                else {
                    pa[py] = px;
                    rel[py] = (rel[mp[l[i] - 1]] - flag - rel[mp[r[i]]] + 4) % 2;
                }
            }
            else {
                if ((rel[mp[l[i] - 1]] - rel[mp[r[i]]] + 2) % 2 != flag) {
                    res = i - 1;
                    break;
                }
            }
        }
        if (res == -1)
            res = q;
        printf("%d\n", res);
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值