NOI2015 程序自动分析

Description

在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。

考虑一个约束满足问题的简化版本:假设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

 

我是咸鱼。转载博客请征得博主同意Orz
原文地址:https://www.cnblogs.com/sdfzhsz/p/9000332.html