BZOJ - 4195 程序自动分析【离散化】【并查集】

【题目描述】
在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本：假设x1,x2,x3,…代表程序中出现的变量，给定n个形如xi=xj或xi≠xj的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。例如，一个问题中的约束条件为：x1=x2，x2=x3，x3=x4，x1≠x4，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。
现在给出一些约束满足问题，请分别对它们进行判定。

【输入】
输入文件的第1行包含1个正整数t，表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题，包含若干行：
第1行包含1个正整数n，表示该问题中需要被满足的约束条件个数。
接下来n行，每行包括3个整数i,j,e，描述1个相等/不等的约束条件，相邻整数之间用单个空格隔开。若e=1，则该约束条件为xi=xj；若e=0，则该约束条件为xi≠xj。

【输出】
输出文件包括t行。
输出文件的第k行输出一个字符串“YES”或者“NO”（不包含引号，字母全部大写），“YES”表示输入中的第k个问题判定为可以被满足，“NO”表示不可被满足。

【样例输入】
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1

【样例输出】
NO
YES

【数据范围及样例解释】
在第一个问题中，约束条件为：x1=x2,x1≠x2。这两个约束条件互相矛盾，因此不可被同时满足。
在第二个问题中，约束条件为：x1=x2,x2=x1。这两个约束条件是等价的，可以被同时满足。
1≤n≤1000000
1≤i,j≤1000000000

题目链接：https://vjudge.net/problem/HYSBZ-4195

离线存下所有条件
对所有等于情况建图，再判断所有不等情况是否满足
数据范围较大，但数据量不大，离散化操作即可

代码如下：

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
static const int MAXN=1e6+10;
struct EQ{
    int i,j;
}eq[MAXN];
struct UEQ{
    int i,j;
}ueq[MAXN];
vector<int> alls;
int f(int x)
{
    int l=0,r=alls.size();
    while(l<r)
    {
        int mid=l+r>>1;
        if(alls[mid]>=x) r=mid;
        else l=mid+1;
    }
    return r+1;
}
int t,n,cnt1,cnt2;
int p[MAXN*2];
void init() {
    for(int i=1;i<=alls.size();i++) p[i]=i;
}
int find(int x) {
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
void merge(int a,int b) {
    p[find(a)]=find(b);
}
int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0),cout.tie(0);
    cin>>t;
    while(t--)
    {
        alls.clear();
        cin>>n;
        cnt1=0,cnt2=0;
        while(n--)
        {
            int i,j,k;
            cin>>i>>j>>k;
            alls.push_back(i);
            alls.push_back(j);
            if(k==1) eq[++cnt1]={i,j};
            else if(k==0) ueq[++cnt2]={i,j};
        }
        sort(alls.begin(),alls.end());
        alls.erase(unique(alls.begin(),alls.end()),alls.end());
        init();
        for(int i=1;i<=cnt1;i++) merge(f(eq[i].i),f(eq[i].j));
        bool flag=true;
        for(int i=1;i<=cnt2;i++)
            if(find(f(ueq[i].i))==find(f(ueq[i].j)))
            {
                flag=false;
                break;
            }
        if(flag) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}