POJ 1703 Find them, Catch them 并查集

原题见POJ 1703

N个人,M种输入,要么是声明两者不是同伙,要么是询问两者是否是同伙。

思路

年幼不懂事,看着紫书上的秩做的,今天重新思考一番,其实原理和积木移动的想法如出一辙。
把人当作积木,可以相互堆叠,a[i]为积木i底下的积木的个数(不严格,也可以是和这个数同奇偶的数字,在每次get累加路径上所有的a才能得到),p[i]为这堆积木最低端的积木编号。初始a[i]=0,p[i]=i.
声明操作:将i所在的积木堆整个移动到j所在的积木堆上方。同时改变a[p[i]]的值,使得get操作之后a[i]和a[j]不同奇偶(不是同伙)。再将p[p[i]]指向p[j]。
询问操作:get操作之后p[i]==p[j]时,i和j才具有可比性(比较的传递性),a[i]和a[j]同奇偶才是同伙,否则不是。

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <queue>
#include <algorithm>
#include <cmath>
using namespace std;
#define N 100100
int n, m, p[N], a[N];
int get(int x){
    if(x != p[x]){
        int t = get(p[x]);
        a[x] += a[p[x]]%2;
        p[x] = t;
    }
    return p[x];
}
int main()
{
    int T;
    scanf("%d", &T);
    while(T--){
        scanf("%d%d", &n, &m);
        char com[10];
        int u, v;
        for(int i = 0;i < n;i++){
            p[i] = i;
            a[i] = 0;
        }
        for(int i = 0;i < m;i++){
            scanf("%s%d%d", com, &u, &v);
            u--; v--;
            get(u); get(v); //先get一下,得到u和v的团长
            if(com[0] == 'A'){
                if(p[u] != p[v]) printf("Not sure yet.\n");
                else if(a[u] % 2 == a[v] % 2) printf("In the same gang.\n");
                else printf("In different gangs.\n");
            }
            else{
                a[p[v]] %= 2;
                a[p[u]] = a[u]%2 == a[v]%2 ? 1:0; //合并的关键
                p[p[u]] = p[v];
            }
        }

    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值