(prog.cpp 1s 256M)
在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。考虑一个约束满足问题的简化版本
:假设x1,x2,x3,…代表程序中出现的变量,给定n个形如xi=xj或xi≠xj的变量相等/不等的约束条件,请判定是否
可以分别为每一个变量赋予恰当的值,使得上述所有约束条件同时被满足。例如,一个问题中的约束条件为:x1=x
2,x2=x3,x3=x4,x1≠x4,这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。现在
给出一些约束满足问题,请分别对它们进行判定。
Input
输入文件的第1行包含1个正整数t,表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题,包含若干行:
第1行包含1个正整数n,表示该问题中需要被满足的约束条件个数。
接下来n行,每行包括3个整数i,j,e,描述1个相等/不等的约束条件,相邻整数之间用单个空格隔开。
若e=1,则该约束条件为xi=xj;若e=0,则该约束条件为xi≠xj。
1≤n≤1000000
1≤i,j≤1000000000
Output
输出文件包括t行。
输出文件的第k行输出一个字符串“YES”或者“NO”(不包含引号,字母全部大写),
“YES”表示输入中的第k个问题判定为可以被满足,“NO”表示不可被满足。
Sample Input
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
Sample Output
NO
YES
HINT
在第一个问题中,约束条件为:x1=x2,x1≠x2。这两个约束条件互相矛盾,因此不可被同时满足。
在第二个问题中,约束条件为:x1=x2,x2=x1。这两个约束条件是等价的,可以被同时满足。
sol:并查集+离散化
先处理相等的关系,把相等关系的数放在同一集合。再处理不相等关系,如果不相等关系的两个数在同一个集合了,说明不满足条件。
离散化的操作,参考https://www.cnblogs.com/cutepota/p/12557997.html。
代码如下:
1 #include<bits/stdc++.h> 2 using namespace std; 3 struct node{ 4 int a,b,e; 5 }p[12500000]; 6 int fa[1252000],flag,cnt,d[1250000],bi[1250000],tmp; 7 int fin(int x) 8 { 9 if (fa[x]==x) 10 return fa[x]; 11 return fa[x]=fin(fa[x]); 12 } 13 int t,n; 14 int main() 15 { 16 scanf("%d",&t); 17 while(t--) 18 { 19 scanf("%d",&n); 20 flag=cnt=tmp=0; 21 memset(d,0,sizeof(d)); 22 memset(bi,0,sizeof(bi)); 23 for(int i=1;i<=n;i++) 24 { 25 scanf("%d%d%d",&p[i].a,&p[i].b,&p[i].e); 26 d[++cnt]=p[i].a; 27 d[++cnt]=p[i].b; 28 } 29 for(int i=1;i<=2*n;i++) 30 fa[i]=i; 31 sort(d+1,d+1+cnt); 32 for(int i=1;i<=cnt;i++) 33 if(i==1||d[i]!=d[i-1]) 34 bi[++tmp]=d[i]; 35 for(int i=1;i<=n;i++)//先处理相等关系 36 if(p[i].e) 37 fa[fin(lower_bound(bi+1,bi+tmp+1,p[i].a)-bi)]=fin(lower_bound(bi+1,bi+1+tmp,p[i].b)-bi); 38 for(int i=1;i<=n;i++)//再处理不相等关系 39 if(!p[i].e) 40 if(fin(lower_bound(bi+1,bi+1+tmp,p[i].b)-bi)==fin(lower_bound(bi+1,bi+1+tmp,p[i].a)-bi)) 41 flag=1; 42 if(flag) 43 printf("NO "); 44 else 45 printf("YES "); 46 } 47 return 0; 48 }