51nod 1515 明辨是非 并查集+set维护相等与不等关系

#include<cstdio>
#include<iostream>
#include<map>
#include<set>
#define R register int
using namespace std;
int n,cnt;
set<int> s[200010];
map<int,int>mp;
int fa[200010];
inline int g() {
    R ret=0,fix=1; register char ch; while(!isdigit(ch=getchar())) fix=ch==‘-‘?-1:fix;
    do ret=ret*10+(ch^48); while(isdigit(ch=getchar())); return ret*fix;
}
int getf(int x) {return x==fa[x]?x:fa[x]=getf(fa[x]);}
signed main() {
    n=g(); for(R i=1;i<=n*2+1;++i) fa[i]=i;
    for(R i=1;i<=n;++i) {
        R x=g(),y=g(),p=g(); 
        if(mp[x]) x=mp[x]; else x=mp[x]=++cnt;
        if(mp[y]) y=mp[y]; else y=mp[y]=++cnt;
        x=getf(x),y=getf(y); if(s[x].size()<s[y].size()) swap(x,y);
        if(p&1) {
            if(s[x].count(y)) putchar(‘N‘),putchar(‘O‘),putchar(‘\n‘);
            else if(x!=y) { fa[y]=x;
                for(set<int>::iterator it=s[y].begin();it!=s[y].end();it++)
                    s[*it].insert(x),s[x].insert(*it);
                putchar(‘Y‘),putchar(‘E‘),putchar(‘S‘),putchar(‘\n‘);
            } else putchar(‘Y‘),putchar(‘E‘),putchar(‘S‘),putchar(‘\n‘);
        } else {
            if(x==y) putchar(‘N‘),putchar(‘O‘),putchar(‘\n‘);
            else s[x].insert(y),s[y].insert(x),
                putchar(‘Y‘),putchar(‘E‘),putchar(‘S‘),putchar(‘\n‘);
        }
    }
}