51 nod D题

这题的时限。。。。无力吐槽。

我的算法和std完全一样。。也不知道为啥会tle（其实有一定概率ac）

不等关系用set存，当限制相等时启发式合并set，然后就没了。

代码：

#include<bits/stdc++.h>
#define rep(i_,a,b) for(int i_=(a);i_<=(b);i_++)

//using namespace std;
using std::map;
using std::set;
const int maxn=100000;
int n,x,y,p,nn,tot;
map<int,int>mmap;
set<int>a[maxn];
int fa[maxn],siz[maxn];
int getfa(int x){
    if (fa[x]==x) return x;
    fa[x]=getfa(fa[x]);
    return fa[x];
}
int main (){
    scanf("%d",&n);
    rep(i,1,n){
        scanf("%d%d%d",&x,&y,&p);
        if (mmap.find(x)==mmap.end()) {
            mmap[x]=++nn;
            fa[nn]=nn;
            siz[nn]=1;
        }
        if (mmap.find(y)==mmap.end()) {
            mmap[y]=++nn;
            fa[nn]=nn;
            siz[nn]=1;
        }
        int xx=mmap[x],yy=mmap[y];
        if (p==0){
               if (getfa(xx)==getfa(yy) ){
                   printf("NO\n");
                }else{
                    printf("YES\n");
                    a[getfa(xx)].insert(getfa(yy));
                    a[getfa(yy)].insert(getfa(xx));
                }
            }else {
            if (getfa(xx)==getfa(yy)){
                printf("YES\n");
            } else if (a[fa[xx]].find(fa[yy])!=a[fa[xx]].end()) {
                printf("NO\n");
            } else {
                printf("YES\n");
                if (siz[fa[xx]]>siz[fa[yy]]) std::swap(xx,yy);
                 siz[fa[yy]]+=siz[getfa(xx)];
                set<int>::iterator t;
                for (t=a[xx].begin();t!=a[xx].end();t++){
                        a[*t].erase(getfa(xx));
                        a[*t].insert(getfa(yy));
                        a[getfa(yy)].insert(*t);
                }
                a[getfa(xx)].clear();
                fa[getfa(xx)]=getfa(yy);
            }
        }
    }
}