poj 1703 - Find them, Catch them(并查集)

思路：

     只要两者的关系确定了，就将他们加入一个集合中，另外增加一个表示关系的数组flag，来表示该节点与其父节点的关系，0表示是同一类，1表示是不同团伙。

     初始时，集合只有自己一个元素。flag设置为0.

代码如下：

const int M = 100005;int flag[M], p[M];int find(int x){    int tmp = p[x];    p[x] = (p[x]==x?x:find(p[x]));    flag[x] = (flag[x]==flag[tmp]?0:1);    return p[x];}char get_char(){    char c;    while(1)    {        c = getchar();        if(c=='A'||c=='D') return c;    }}int main(){    int t, n, m, a, b, c, x, y;    scanf("%d", &t);    while(t--)    {        scanf("%d%d", &n, &m);        memset(flag,0,sizeof(flag));        for(int i = 0; i <= n; ++i) p[i] = i;        for(int i = 0; i < m; ++i)        {            c = get_char();            scanf("%d %d", &a, &b);            x = find(a);            y = find(b);            if(c=='A')            {                if(x!=y) puts("Not sure yet.");                else if(flag[a]==flag[b]) puts("In the same gang.");                else puts("In different gangs.");            }            else            {                if(x!=y)                {                    p[x] = y;                    flag[x] = (flag[a]==flag[b]?1:0);                }            }        }    }    return 0;}