【51nod 1515】 明辨是非

description

给n组操作，每组操作形式为x y p。

当p为1时，如果第x变量和第y个变量可以相等，则输出YES，并限制他们相等；否则输出NO，并忽略此次操作。

当p为0时，如果第x变量和第y个变量可以不相等，则输出YES，并限制他们不相等 ；否则输出NO，并忽略此次操作。

Solution

对于p为0的情况，我们用并查集判断一下即可。

那么现在来处理p为1的情况。

我们对于每个相等集合用一个set存储和它不相等的数分别有哪些。

当p=0时，我们将x和y分别加入对方的集合的set内。

当p=1时，我们只用判断一下对方集合内是否有自己即可。然后将两个集合合并时，只要将原有集合的set转移到另一个集合的set即可。注意：由于这个set是双向的(即若x和y不相等，那么x和y会分别加入对方的集合的set内)，所以，原有集合的set内的数t的set也要修改一下。

Code

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<set>
using namespace std;
const int maxn=2*1e5+6;
set<int> a[maxn];
struct code{
    int a,b;
}b[maxn];
int fa[maxn],size[maxn],c[maxn];
int n,i,t,j,k,l,x,y,num,xx,yy;
bool cmp(code x,code y){
    return x.a<y.a;
}
int getfather(int x){
    if (x==fa[x]) return x;
    fa[x]=getfather(fa[x]);return fa[x];
}
int main(){
    //freopen("data.in","r",stdin);freopen("data.out","w",stdout);
    scanf("%d",&n);
    for (i=1;i<=n;i++)
        scanf("%d%d%d",&b[i].a,&b[i+n].a,&c[i]),b[i].b=b[i+n].b=i;
    sort(b+1,b+2*n+1,cmp);
    for (i=1;i<=2*n;i++){
        if (b[i].a!=x) ++num;
        x=b[i].a,b[i].a=num;
        swap(b[i].a,b[i].b);
    }
    sort(b+1,b+2*n+1,cmp);
    for (i=1;i<=num;i++)
        fa[i]=i;
    for (i=1;i<=n;i++){
        x=b[i*2-1].b;
        y=b[i*2].b;
        xx=getfather(x);yy=getfather(y);
        if (!c[i]){
            if (xx==yy) printf("NO\n");
            else{
                a[xx].insert(yy);
                a[yy].insert(xx);  
                printf("YES\n");
            }
        }else{
            if (a[xx].count(yy)) printf("NO\n");
            else if (xx==yy) printf("YES\n");
            else{
                if (a[xx].size()>a[yy].size()) swap(xx,yy);
                while (!a[xx].empty()){
                    t=*--a[xx].end();
                    a[yy].insert(t);
                    a[t].erase(xx);
                    a[t].insert(yy); 
                    a[xx].erase(t); 
                }
                fa[xx]=yy;
                printf("YES\n");
            }
        }
    }
}