程序自动分析

(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 }
原文地址:https://www.cnblogs.com/cutepota/p/12594884.html