poj 1703 Find them, Catch them 带权并查集

题目：http://poj.org/problem?id=1703

题意：在这个城市里有两个黑帮团伙，现在给出N个人，问任意两个人他们是否在同一个团伙
输入D x y代表x于y不在一个团伙里
输入A x y要输出x与y是否在同一团伙或者不确定他们在同一个团伙里

思路：带权并查集。定义rnk[x]为x和x的父亲par[x]之间的关系，同一个帮派为0，不同帮派为1。设x是y的儿子，y是z的儿子，用穷举法可以发现x和z的关系可以表示为(rnk[x] + rnk[y]) % 2，如果z还有祖先的时候，应当先确定y和祖先的关系，再确定x和祖先的关系。当两个点x y属于不同组进行合并时，设它们祖先分别为fx fy(假设所有的树只有两层，实际上路径压缩后必然只有两层)，可以先求出fy相对于y的关系，然后就可以依次求出fy相对于x的关系、相对于fx的关系，写成一个式子就是(rnk[y] + type + rnk[x]) % 2（显然rnk[fy]=rnk[y]，type是y和x之间的关系），然后把fy的父亲设为fx。判断x y关系时，祖先不同时必然是Not sure yet.，相同时，可以求出它们的父亲f和x的关系，也就是rnk[x]，然后rnk[y]+rnk[x]就是y和x的关系。下面还贴了一份别人的代码，很巧妙。另外挑战程序设计上也有一个很好的思路，种类并查集的思路

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;


const int N = 100000 + 10, INF = 0x3f3f3f3f;


int par[N], rnk[N];


void init(int n)
{
    for(int i = 1; i <= n; i++) par[i] = i, rnk[i] = 0;
}
int ser(int x)
{
    if(par[x] != x)
    {
        int fx = ser(par[x]);
        rnk[x] = (rnk[x] + rnk[par[x]]) % 2;
        par[x] = fx;
    }
    return par[x];
}
void unite(int x, int y, int type)
{
    int fx = ser(x), fy = ser(y);
    if(fx == fy) return;
    rnk[fy] = (rnk[y] + type + rnk[x]) % 2;
    par[fy] = fx;
}
int check(int x, int y)
{
    int fx = ser(x), fy = ser(y);
    if(fx != fy) return 0;
    else
    {
        if((rnk[y] + rnk[x]) % 2 == 0) return 2;
        else return 1;
    }
}
int main()
{
    int t, n, m;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &m);
        init(n);
        int x, y;
        char ch;
        for(int i = 1; i <= m; i++)
        {
            scanf(" %c%d%d", &ch, &x, &y);
            if(ch == 'A')
            {
                int res = check(x, y);
                if(res == 0) printf("Not sure yet.\n");
                else if(res == 1) printf("In different gangs.\n");
                else if(res == 2) printf("In the same gang.\n");
            }
            else unite(x, y, 1);
        }
    }
    return 0;
}

还看到另外巧妙的做法，一起贴上来

/*把节点的对立点存到op数组中，再次碰到和当前点对立的点，则一定和op数组中储存的点是同一集合*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 100100;
int n, m;
int par[N], op[N];
void init()
{
    for(int i = 1; i <= n; i++)
        par[i] = i, op[i] = 0;
}
int ser(int x)
{
    int r = x;
    while(r != par[r]) r = par[r];
    int i = x, j;
    while(i != r) j = par[i], par[i] = r, i = j;
    return r;
}
void unite(int x, int y)
{
    x = ser(x), y = ser(y);
    if(x == y) return;
    par[x] = y;
}

int main()
{
    int t, b, c;
    char a;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d%d", &n, &m);
        init();
        for(int i = 0; i < m; i++)
        {
            scanf(" %c%d%d", &a, &b, &c);
            if(a == 'D')
            {
                if(op[b] == 0 && op[c] == 0) op[b] = c, op[c] = b;
                else if(op[b] == 0) op[b] = c, unite(b, op[c]);
                else if(op[c] == 0) op[c] = b, unite(c, op[b]);
                else unite(b, op[c]), unite(c, op[b]);
            }
            else
            {
                if(n == 2) printf("In the same gang.\n");
                else if(ser(b) == ser(c)) printf("In the same gang.\n");
                else if(ser(b) == ser(op[c])) printf("In different gangs.\n");
                else printf("Not sure yet.\n");
            }
        }
    }
    return 0;
}