并查集（应用）

并查集的应用（补充）

• 并查集的基础操作与介绍请看上一篇博客并查集

• 北大POJ平台705题，“发现他，抓住他”。

• 描述

  一个城市中有两个犯罪团伙A和B，你需要帮助警察判断任意两起案件是否是同一个犯罪团伙所为，警察所获得的信息是有限的。假设现在有N起案件（N<=100000），编号为1到N，每起案件由团伙A或团伙B所为。你将按时间顺序获得M条信息（M<=100000），这些信息分为两类：
  1 、D[a][b]
  其中[a]和[b]表示两起案件的编号，这条信息表明它们属于不同的团伙所为
  2、 A [a] [b]
  其中[a]和[b]表示两起案件的编号，这条信息需要你回答[a]和[b]是否是同一个团伙所为

  注意你获得信息的时间是有先后顺序的，在回答的时候只能根据已经接收到的信息做出判断。

  输入

  第一行是测试数据的数量T（1<=T<=20）。
  每组测试数据的第一行包括两个数N和M，分别表示案件的数量和信息的数量，其后M行表示按时间顺序收到的M条信息。

  输出

  对于每条需要回答的信息，你需要输出一行答案。如果是同一个团伙所为，回答"In the same gang."，如果不是，回答"In different gangs."，如果不确定，回答”Not sure yet."。

  样例输入

  1
  5 5
  A 1 2
  D 1 2
  A 1 2
  D 2 4
  A 1 4
  

  样例输出

  Not sure yet.
  In different gangs.
  In the same gang.
  

• 解析：

  • 本题与传统并查集的区别在于本题给出了不属于一个集合的元素，根据只有两个集合来反推出属于一个集合的元素，所以需要在传统的给出谁属于一个集合的并查集上进行改动。具体改动注释于代码中。

#include<iostream>
#include<cstdio>
using namespace std;

int fx[100010];
int flag[100010];//增加标志位，标志位为1和0，用于判断两者是否属于同一组。
void initial1(int n){
    for(int i=1;i<=n;i++){
        fx[i]=i;//与原本相同，不做改变。每个元素初始都是一个集合。
        flag[i]=0;//初始值赋零。
    }
}

int find1(int a){
    int t=0;
    if(fx[a]==a){
        return a;
    }else{
        t = find1(fx[a]);//路径压缩
        flag[a] = (flag[fx[a]] +flag[a]) % 2;//信息给出的是两个不属于同一个集合的元素，当我们把其合并后，因”相邻“的（在这里称某个元素与其父节点为相邻，除其父节点外都不相邻）两节点的标志信息肯定不同，我们根据父节点的标志位对子节点的标志位进行修改。此处修改可以这样理解：union操作已经设置好了标志位的信息，但是路径压缩时，以祖、父、子三个节点为例，我们要把子节点改接到祖先节点上，因为祖先节点与父节点“相邻”，两者标志位不同，父节点与子节点标志位有又不同，所以子节点肯定与祖先节点相同，此时我们路径压缩时就需要更具父节点再次修改子节点的标志信息。
        fx[a] = t;
        return fx[a];
    }
}

void union1(int a,int b){
    int p = find1(a);
    int q = find1(b);
    fx[p] = q;//合并操作，需要小修改，因为find操作进行了路径压缩，修改了fx,不能fx[find(a)]=find(b);这样修改。
    if(flag[b] == 0)//此处为合并操作时对标志信息的设置，因a,b两节点合并后就是“相邻”的父子节点，其标志位要相反，所以根据其一修改另一个的标志位。
        flag[p] = 1-flag[a];
    else
        flag[p] = flag[a];

}
//用scanf输入信息，cin可能会超时。
int main(){
    int t,n,m,x,y;
    char ch;
    scanf("%d",&t);
    while(t--){
         scanf("%d %d",&n,&m);
        initial1(n);
        while (m--)
        {
            getchar();//此处
            scanf("%c %d %d",&ch,&x,&y);
            if(ch=='A'){
                if(find1(x)!=find1(y)){
                    cout<<"Not sure yet."<<endl;
                }
                else if(flag[x]==flag[y]){
                    cout<<"In the same gang."<<endl;
                }
                else {
                    cout<<"In different gangs."<<endl;
                }
            }
            else if(ch=='D'){
                union1(x,y);
            }
        
        }

    }
    system("pause");
    return 0;
    
}