例题来源,牛客网

程序自动分析

链接:https://ac.nowcoder.com/acm/contest/1031/A
来源:牛客网

题目描述

在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本:假设 𝑥1, 𝑥2, 𝑥3, ⋯ 代表程序中出现的变量,给定 𝑛 个形如 𝑥𝑖 = 𝑥𝑗 或 𝑥𝑖 ≠ 𝑥𝑗 的变量
相等/不等的约束条件,请判定是否可以分别为每一个变量赋予恰当的值,使得上述所有约束条件同时被满足。例如,一个
问题中的约束条件为: 𝑥1 = 𝑥2, 𝑥2 = 𝑥3, 𝑥3 = 𝑥4, 𝑥1 ≠ 𝑥4 ,这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。
现在给出一些约束满足问题,请分别对它们进行判定。

输入描述:

第1行包含1个正整数t,表示需要判定的问题个数。注意这些问题之间是相互独立的。

对于每个问题,包含若干行:
第1行包含1个正整数n,表示该问题中需要被满足的约束条件个数。
接下来n行,每行包括3个整数i,j,e,描述1个相等/不等的约束条件,相邻整数之间用单个空格隔开。
若e=1,则该约束条件为xi=xj;若e=0,则该约束条件为xi≠xj。

输出描述:

包括t行。

第k行输出一个字符串“YES”或者“NO”(不包含引号,字母全部大写),“YES”表示输入中的第k个问题判定为可以被满足,“NO”表示不可被满足。

输入

2
2
1 2 1
1 2 0
2
1 2 1
2 1 1

输出

NO
YES

 1>首先,很容易能够看出来是一道可以用并查集做出来的题目,那么就剩一个问题了,就是数据过大,
那么就需要进行数据离散化处理,具体如何离散化的请参照代码

 2>其次,需要离线的对数据进行合并以及查错,所以需要把初始的数据用结构体数组保存起来,
如果在线处理的话可能会造成后面的合并条件影响了前面的值,使得前面的结果不准确;

3> 最后,需要把所有e==1的点合并Merge,然后在处理e==0的情况,此时如果出现x,y的两个点的根为一个那么就可以判定出错,
就可以不用再进行下面的运算


代码如下:
 1 #include <cstdio>
 2 #include <algorithm>
 3 using namespace std;
 4 
 5 struct exp{
 6     int i,j,e;
 7 }arr[100005];
 8 int a[200010],f[200010];
 9 
10 inline void Init(int cnt){
11     for(int i=1;i<=cnt;i++)
12         f[i]=i;
13 }
14 int find(int x){
15     if(x==f[x])
16         return x;
17     int fx=find(f[x]);
18     f[x]=fx;
19     return f[x];
20 }
21 void Merger(int x,int y){
22     int fa=find(x);
23     int fb=find(y);
24     if(fa!=fb)
25         f[fa]=fb;
26 }
27 
28 int main(void)
29 {
30     int t;
31     scanf("%d",&t);
32 
33     int n,x,y,z,cnt;
34     while(t--){
35         cnt=0;
36         scanf("%d",&n);
37         for(int i=1;i<=n;i++){
38             scanf("%d%d%d",&x,&y,&z);
39             a[++cnt]=x;
40             a[++cnt]=y;
41             arr[i]={x,y,z};
42         }
43         sort(a+1,a+1+cnt);   //将所有的点读入,再进行排序,为后面的离散化做准备
44         cnt=unique(a+1,a+1+cnt)-(a+1);//去掉重复的数,同时改变cnt的值
45         Init(cnt);
46         for(int i=1;i<=n;i++){
47             arr[i].i=lower_bound(a+1,a+1+cnt,arr[i].i)-a;  //找到离散化后的arr[i].i对应的数值
48             arr[i].j=lower_bound(a+1,a+1+cnt,arr[i].j)-a;
49         }
50         for(int i=1;i<=n;i++){
51             if(arr[i].e==1)
52                 Merger(arr[i].i,arr[i].j);
53         }
54         int flag=0;
55         for(int i=1;i<=n;i++){
56             if(!arr[i].e){
57                 if(find(arr[i].i)==find(arr[i].j)){
58                     flag=1;
59                     break;
60                 }
61             }
62         }
63         if(flag)
64             printf("NO\n");
65         else
66             printf("YES\n");
67     }
68     return 0;
69 }