【并查集+离散化】BZOJ4195- [Noi2015]程序自动分析

【题目大意】

 在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。

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

现在给出一些约束满足问题，请分别对它们进行判定。

【思路】

智障啊！！

完全是一道超级水的并查集，我居然以为是和关押罪犯一样分为两类的……

简单来说，先把放一起的处理掉，然后再判断不能放在一起的，如果它们在一个并查集里着返回“No”...

忘掉unique怎么写了…………按照自己的想法胡乱地弄了一下离散化……胡乱地……

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector> 
using namespace std;
const int MAXN=2000000+500;
struct node
{
    int num;
    int oripos;
    int ab;//0a1b
    bool operator < (const node &x) const
    {
        return num<x.num;
    }
}que[MAXN];
int h[MAXN],u[MAXN];
int n,a[MAXN],b[MAXN],e[MAXN];

void union_set(int a,int b)
{
    if (h[a]>=h[b])
    {
        u[b]=a;
        if (h[a]==h[b]) h[a]++;
    }
    else u[a]=b;
}

int find(int x)
{
    int r=x;
    while (u[r]!=r) r=u[r];
    int p=x;
    while (u[p]!=p)
    {
        int tmp=u[p];
        u[p]=r;
        p=tmp;
    }
    return r;
}


void init()
{
    scanf("%d",&n);
    int qlen=-1;
    for (int i=0;i<n;i++)
    {
        scanf("%d%d%d",&a[i],&b[i],&e[i]);
        que[++qlen].num=a[i];
        que[qlen].oripos=i;
        que[qlen].ab=0;
        que[++qlen].num=b[i];
        que[qlen].oripos=i;
        que[qlen].ab=1;
    } 
    sort(que,que+2*n);
    int j=0;
    for (int i=0;i<2*n;i++)
    {
        if (i!=0 && que[i].num!=que[i-1].num) j++;
        if (que[i].ab==0) a[que[i].oripos]=j;
            else b[que[i].oripos]=j;
    }
    memset(h,0,sizeof(h));
    for (int i=0;i<=2*(n+2);i++) u[i]=i;
}

int judge()
{
    for (int i=0;i<n;i++)
        if (e[i]) 
        {
            union_set(find(a[i]),find(b[i]));
        }
    for (int i=0;i<n;i++)
        if (!e[i])
        {
            int fa=find(a[i]),fb=find(b[i]);
            if (fa==fb) return 0;
        }
    return 1;
}

int main()
{
    int T;
    scanf("%d",&T);
    while (T--)
    {
        init();
        int j=judge();
        if (j) cout<<"YES"<<endl;else cout<<"NO"<<endl;
    }
    return 0;
}