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;
}