题目链接:http://poj.org/problem?id=1703
题目:
警方决定捣毁两大犯罪团伙:龙帮和蛇帮,显然一个帮派至少有一人。该城有N个罪犯,编号从1至N(N<=100000。将有M(M<=100000)次操作。
D a b 表示a、b是不同帮派
A a b 询问a、b关系
Input
多组数据。第一行是数据总数 T (1 <= T <= 20)每组数据第一行是N、M,接下来M行是操作
Output
对于每一个A操作,回答”In the same gang.”或”In different gangs.” 或”Not sure yet.”
Sample Input
1
5 5
A 1 2
D 1 2
A 1 2
D 2 4
A 1 4
思路见代码:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
using namespace std;
const int maxn = 100000+5;
int fa[2*maxn],height[maxn*2];//height数组用来控制并查的方向
int T,N,M,a,b;
char c;
void init()//初始化
{
for(int i=1; i<=2*N; i++)
{
fa[i]=i;
height[i]=0;
}
}
int find_(int x)//并查集
{
return x==fa[x]? x : fa[x]=find_(fa[x]);
}
void unit (int u, int v)
{
int x,y;
x=find_(u);
y=find_(v);
if(x==y)
return;
if(height[y]>height[x])
fa[x]=y;
else
{
fa[y]=x;
if(height[x]==height[y])
height[x]++;
}
}
bool same(int u, int v)
{
return find_(u)==find_(v);
}
int main()
{
scanf("%d",&T);
while(T--)
{
flag=0;
scanf("%d %d",&N,&M);
init();
for(int i=0; i<M; i++)
{
getchar();
scanf("%c %d %d",&c,&a,&b);
if(c=='D')
{
unit(a,b+N);// b和b+N不在同一个帮派,则把a和b+N看作一个帮派的,合并得到一起
unit(a+N,b);// 同理由上可得
}
else
{
if(same(a,b))
printf("In the same gang.\n");
else if(same(a,b+N))
printf("In different gangs.\n");
else
printf("Not sure yet.\n");
}
}
}
return 0;
}