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 + 5;

int n, cnt = 1,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 +=5;
    for (int i = 1; i <= n; i++)
    {
        f[i] = i;
    }
    n -=5;
    
    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 == 1)//限制相等 
        {
            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 0;
}