poj 1703——Find them, Catch them

题意：有两个帮派，警察要调查出每个人属于哪个帮派。每次更新的两个数据是属于不同帮派的人，每次查询的是两个人是否属于一个帮派。
思路：用并查集来表示可以扯上关系的人。如果两个人的关系可以判定，那么他们就属于一个集合。在一个集合中，用r来表示节点跟父节点的距离。如果距离是0，那么这个节点就是父节点。如果距离是1，那么这个节点跟父节点相差一个，就属于不同帮派，如果距离是2，就是一个帮派，以此类推。

#include<iostream>#include<cstdio>#include<cstring>using namespace std;const int maxn=100005;int father[maxn];int r[maxn];int n,m;void init(){        for(int i=1;i<=n;++i){                father[i]=i;                r[i]=0;        }}int find(int x){        if(x==father[x]){                return x;        } else {                int tmp=father[x];                father[x]=find(father[x]);                r[x]=r[x]+r[tmp];                return father[x];        }}void merge(int a,int b){        int x=find(a);        int y=find(b);        if(x!=y){                father[x]=b;                r[x]=r[a]+1;        }}void solve(int a,int b){        int x=find(a);        int y=find(b);        if(x==y){                if((r[a]+r[b])&1){                        puts("In different gangs.");                } else {                        puts("In the same gang.");                }        } else {                printf("Not sure yet.\n");        }}int main(){//        freopen("data.txt","r",stdin);        int T;        scanf("%d",&T);        while(T--){                scanf("%d%d",&n,&m);                init();                for(int i=0;i<m;++i){                        char op='z';                        while(op!='A'&&op!='D'){                                scanf("%c",&op);                        }                        int a,b;                        scanf("%d%d",&a,&b);                        if(op=='A'){                                solve(a,b);                        } else {                                merge(a,b);                        }                }        }        return 0;}