poj1703--Find them, Catch them

题意：一个城市n个犯罪嫌疑人，编号1-n，每次输入D x y表示x y属于同一帮派，A x y询问x y是否同一帮派或者不确定。

带权、种类并查集裸题，图片质量不好还请见谅。。oet[fx] = (oet[y]-oet[x]+d+2)%2   根据箭头关系就可以得出这个式子了，，加2是防止出现负值

 

#include <set>

#include <map>

#include <cmath>

#include <ctime>

#include <queue>

#include <stack>

#include <cctype>

#include <cstdio>

#include <string>

#include <vector>

#include <cstdlib>

#include <cstring>

#include <iostream>

#include <algorithm>

using namespace std;

typedef unsigned long long ull;

typedef long long ll;

const int inf = 0x3f3f3f3f;

const double eps = 1e-8;

const int maxn =1e5+10;

int n,m;

int  pa[maxn],oet[maxn];

void init()

{

    for (int i = 0; i < maxn; i++)

    {

        pa[i] = i;

    }

    memset(oet,0,sizeof(oet));

}

int find(int x)

{

    if (pa[x] == x)

    {

        return x;

    }

    else

    {

        int tmp = pa[x];

        pa[x] = find(tmp);

        oet[x] = (oet[x]+oet[tmp])%2;

        return pa[x];

    }

}

void union_set(int x,int y,int d)

{

    int fx = find(x);

    int fy = find(y);

    if (fx == fy)

        return;

    pa[fx] = fy;

    oet[fx] = (oet[y]-oet[x]+d+2)%2;//oet[x] 表示x对其父亲的偏移量。至于这个式子怎么退出的，，就用向量来推，起点到终点

}

void query(int x,int y)

{

    int fa = find(x);

    int fy = find(y);

    if (fa != fy)                       //x y不在一个集合中自然不能确定关系。

    {

         printf("Not sure yet.\n");

         return;

    }

    int tmp = (oet[x]-oet[y]+2)%2;       //在求这种关系式的时候可以借助数学的向量来推导。

    if (tmp==1)

        printf("In different gangs.\n");

    else

        printf("In the same gang.\n");

}

int main(void)

{

    int t;

    cin>>t;

    while (t--)

    {

        scanf("%d%d",&n,&m);

        char ch[10];

        int x,y;

        init();

        for (int i = 0; i < m; i++)

        {

            scanf("%s%d%d",ch,&x,&y);

            if (ch[0]=='A')

            {

                query(x,y);

            }

            else

            {

                union_set(x,y,1);

            }

        }

    }

    return 0;

}