51nod 1515 明辨是非[并查集][set]

对于每一个数分别维护并查集和一个集合
并查集用于判断相等关系
集合用于判断不等关系:每个数的集合中原来为空,如果两个数 x,y 被要求不等就将它们互相插入到集合中,然后这个可以用set实现,按集合(set)中的元素个数大小进行合并(按秩合并)
若p=1,先判断是否在同一个并查集中,若不在则到set中找
若p=0,先判断是否在同一个并查集中,若不在互相插入

#include<cstdio>#include<algorithm>#include<set>#include<map>#define N 200000using namespace std;inline char nc(){    static char buf[100000],*p1=buf,*p2=buf;    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;}inline void read(int &x){    char c=nc(),b=1;    for(;!(c>='0'&&c<='9');c=nc()) if(c=='-') b=-1;    for(x=0;c>='0'&&c<='9';x=x*10+c-'0',c=nc()); x*=b;}int n,cnt=1,fa[N+5];set<int>s[N+5];map<int,int>m;inline int Pos(int x){return m[x]?m[x]:(m[x]=++cnt);}int getf(int x){return fa[x]==x?x:fa[x]=getf(fa[x]);}int main(){    read(n);    for(int i=1;i<=n*2;++i) fa[i]=i;    int x,y,opt;    while(n--)    {        read(x),read(y),read(opt);        x=Pos(x),y=Pos(y);        int fx=getf(x),fy=getf(y);        if(opt==1)        {            if(s[fx].count(fy)) puts("NO");            else if(fx!=fy)            {                if(s[fx].size()>s[fy].size()) swap(fx,fy);                fa[fx]=fy;                for(set<int>::iterator pos=s[fx].begin();pos!=s[fx].end();++pos)                    s[*pos].erase(fx),s[*pos].insert(fy),s[fy].insert(*pos);                puts("YES");            }            else puts("YES");        }        else        {            if(fx==fy) puts("NO");            else            {                s[fx].insert(fy);                s[fy].insert(fx);                puts("YES");            }        }    }    return 0;}