【并查集+离散化】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=+;
 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=-;
     for (int i=;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=;
         que[++qlen].num=b[i];
         que[qlen].oripos=i;
         que[qlen].ab=;
     }
     sort(que,que+*n);
     int j=;
     for (int i=;i<*n;i++)
     {
         if (i!= && que[i].num!=que[i-].num) j++;
         if (que[i].ab==) a[que[i].oripos]=j;
             else b[que[i].oripos]=j;
     }
     memset(h,,sizeof(h));
     for (int i=;i<=*(n+);i++) u[i]=i;
 }

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

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