POJ1703 - Find them, Catch them

原题链接

Description

一个城市中有$n(n\leq10^5)$个人，他们分别属于两个帮派。给出$m(m\leq10^5)$次操作：
- 告诉你$x$与$y$分属不同帮派。
- 问$x$与$y$的关系。关系有三种：帮派相同，帮派不同，不确定。

Solution

并查集。$e[x]$表示$x$的一个敌人，初始时为0。每次操作1时，将$x$加入$e[y]$的集合，$y$加入$e[x]$的集合。对于操作2：
- 当且仅当$x$与$y$同属一个集合时帮派相同；
- 当且仅当$x$与$e[y]$同属一个集合时帮派不同；
- 其余情况无法确定。

时间复杂度$O(m)$。

Code

//Find them, Catch them
#include <cstdio>
#include <cstring>
inline int read()
{
    int x=0; char ch=getchar();
    while(ch<'0'||'9'<ch) ch=getchar();
    while('0'<=ch&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
int const N=1e5+10;
int n,m;
int fa[N],e[N];
int find(int u) {return fa[u]==u?u:fa[u]=find(fa[u]);}
int main()
{
    int task=read();
    while(task--)
    {

    n=read(),m=read();
    for(int i=0;i<=n;i++) fa[i]=i,e[i]=0;
    for(int i=1;i<=m;i++)
    {
        char opt=getchar(); while(opt!='A'&&opt!='D') opt=getchar();
        int x=read(),y=read();
        if(opt=='D')
        {
            if(!e[x]) e[x]=y; if(!e[y]) e[y]=x;
            fa[find(x)]=find(e[y]); fa[find(y)]=find(e[x]);
        }else{
            if(find(x)==find(y)) printf("In the same gang.\n");
            else if(find(x)==find(e[y])) printf("In different gangs.\n");
            else printf("Not sure yet.\n");
        }
    }

    }
    return 0;
}

P.S.

原来以前写的并查集都是错的…不是fa[find(x)]=y而是fa[find(x)]=find(y)。前者在$x$与$y$同属一个集合时会成环。