51nod-1515 明辨是非——并查集

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

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

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

Input

输入一个数n表示操作的次数(n<=1*10^5)
接下来n行每行三个数x,y,p（x,y<=1*10^8,p=0 or 1）

Output

对于n行操作，分别输出n行YES或者NO

Input示例

3
1 2 1
1 3 1
2 3 0

Output示例

YES
YES
NO

这题目描述实在是很迷，看了半天才明白，一共三种关系，相等，不相等和不确定，只有YES操作会改变不确定关系。
用set来维护限制不相等的数，用并查集维护限制相等的数，两个数从不确定到不相等，互相插入对方的set；两个数从不确定到相等，改变他们的父结点，默认将小的那个集合的父结点改成大的，就是大吃小的意思。

 #include <iostream>
 #include <cstdio>
 #include <set>
 #include <map>

 using namespace std;

 const int MAXN = 2e5 + ;

 int n, cnt = ,f[MAXN];
 set<int> nt[MAXN];
 map<int, int> eq;

 int find(int x)
 {
     return f[x] == x ? x : f[x] = find(f[x]);
 }

 int main()
 {
     //freopen("in.txt","r",stdin);
     cin>>n;
     n +=;
     for (int i = ; i <= n; i++)
     {
         f[i] = i;
     }
     n -=;

     int x, y, p;
     while (n--)
     {
         cin>>x>>y>>p;
         if (eq[x])//计数标号
         {
             x = eq[x];
         }
         else
         {
             ++cnt;
             eq[x]=cnt;
             x=cnt;
         }
         if (eq[y])
         {
             y = eq[y];
         }
         else
         {
             ++cnt;
             eq[y]=cnt;
             y=cnt;
         }
         int u = find(x), v = find(y);
         if (p == )//限制相等
         {
             if (nt[u].count(v)) //限制不相等
             {
                 cout<<"NO"<<endl;
             }
             else if (u != v)    //未限制但并未相等
             {
                 if (nt[u].size() > nt[v].size())
                 {
                     swap(u, v);
                 }
                 f[u] = v;
                 set<int>::iterator it;
                 for (it = nt[u].begin(); it != nt[u].end(); it++)
                 {
                     nt[*it].insert(v);
                     nt[v].insert(*it);
                 }
                 cout<<"YES"<<endl;
             }
             else
             {
                 cout<<"YES"<<endl;
             }
         }
         else//限制不相等
         {
             if (u == v)
             {
                 cout<<"NO"<<endl;
             }
             else
             {
                 nt[u].insert(v);
                 nt[v].insert(u);
                 cout<<"YES"<<endl;
             }
         }
     }

     return ;
 }