在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本:假设x1,x2,x3...代表程序中出现的变量,给定n个形如xi=xj或xi≠xj的变量相等/不等的约束条件,请判定是否可以分别为每一个变量赋予恰当的值,使得上述所有约束条件同时被满足。例如,一个问题中的约束条件为:x1=x2,x2=x3,x3=x4,x4≠x1,这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。
现在给出一些约束满足问题,请分别对它们进行判定。
Input:
从文件prog.in中读入数据。
输入文件的第1行包含1个正整数t,表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题,包含若干行:
第1行包含1个正整数n,表示该问题中需要被满足的约束条件个数。接下来n行,每行包括3个整数i,j,e,描述1个相等/不等的约束条件,相邻整数之间用单个空格隔开。若e=1,则该约束条件为xi=xj;若e=0,则该约束条件为xi≠xj;
Output:
输出到文件 prog.out 中。
输出文件包括t行。
输出文件的第 k行输出一个字符串“ YES” 或者“ NO”(不包含引号,字母全部大写),“ YES” 表示输入中的第k个问题判定为可以被满足,“ NO” 表示不可被满足。
Solution
本题是个简单的并查集,但是因为数据过大,需要用到离散化,先把数据的询问排个序,然后先处理相等的,最后在处理不等关系的时候,如果发现他们在一个相等的并查集中,就代表不能满足,因此就输出"NO",如果完整的走下来,就输出"YES"即可
1 //Writer : Hsz %WJMZBMR%tourist%hzwer 2 #include<iostream> 3 #include<cstring> 4 #include<cstdio> 5 #include<cmath> 6 #include<queue> 7 #include<map> 8 #include<set> 9 #include<stack> 10 #include<vector> 11 #include<cstdlib> 12 #include<algorithm> 13 #define LL long long 14 using namespace std; 15 int T,n,x,y,z; 16 int fa[200005]; 17 struct Q { 18 int a,b,c; 19 } q[100005]; 20 int e[200005]; 21 int find(int x) { 22 if(x==fa[x]) return x; 23 return fa[x]=find(fa[x]); 24 } 25 bool cmp(Q x,Q y) { 26 return x.c>y.c; 27 } 28 int main() { 29 cin>>T; 30 while(T--) { 31 scanf("%d",&n); 32 int cnt=0; 33 for(int i=1; i<=n; i++) { 34 scanf("%d%d%d",&q[i].a,&q[i].b,&q[i].c); 35 e[++cnt]=q[i].a; 36 e[++cnt]=q[i].b; 37 } 38 bool flag=0; 39 sort(q+1,q+1+n,cmp); 40 sort(e+1,e+1+cnt); 41 int u=unique(e+1,e+1+cnt)-e-1; 42 for(int i=0; i<=u; i++) fa[i]=i; 43 for(int i=1; i<=n; i++) { 44 q[i].a=lower_bound(e+1,e+1+u,q[i].a)-e;/*离散化过程。 45 注意lower_bound的时候 46 加的是unique后的数组大小*/ 47 q[i].b=lower_bound(e+1,e+1+u,q[i].b)-e; 48 if(q[i].c) { 49 fa[find(q[i].a)]=find(q[i].b); 50 } else { 51 if(find(q[i].a)==find(q[i].b)) { 52 puts("NO"); 53 flag=1; 54 break; 55 } 56 } 57 } 58 if(!flag) puts("YES"); 59 } 60 return 0; 61 } 62