P1955 [NOI2015] 程序自动分析

题目链接

luogu P1955

题面

题目描述

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

考虑一个约束满足问题的简化版本：假设 $x_1,x_2,x_3,\cdots$ 代表程序中出现的变量，给定 $n$ 个形如 $x_i=x_j$ 或 $x_i\ne x_j$ 的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。例如，一个问题中的约束条件为：$x_1=x_2,x_2=x_3,x_3=x_4,x_4\ne x_1$，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

现在给出一些约束满足问题，请分别对它们进行判定。

输入格式

输入的第一行包含一个正整数 $t$，表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题，包含若干行：
第一行包含一个正整数 $n$，表示该问题中需要被满足的约束条件个数。接下来 $n$ 行，每行包括三个整数 $i,j,e$，描述一个相等/不等的约束条件，相邻整数之间用单个空格隔开。若 $e=1$，则该约束条件为 $x_i=x_j$。若$e=0$，则该约束条件为 $x_i\ne x_j$。

输出格式

输出包括 $t$ 行。
输出文件的第 $k$ 行输出一个字符串 YES 或者 NO（字母全部大写），YES 表示输入中的第 $k$ 个问题判定为可以被满足，NO 表示不可被满足。

样例 #1

样例输入 #1

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

样例输出 #1

NO
YES

样例 #2

样例输入 #2

2
3
1 2 1
2 3 1
3 1 1
4
1 2 1
2 3 1
3 4 1
1 4 0

样例输出 #2

YES
NO

提示

【样例解释1】

在第一个问题中，约束条件为：$x_1=x_2,x_1\ne x_2$。这两个约束条件互相矛盾，因此不可被同时满足。
在第二个问题中，约束条件为：$x_1=x_2,x_1=x_2$。这两个约束条件是等价的，可以被同时满足。

【样例说明2】

在第一个问题中，约束条件有三个：$x_1=x_2,x_2=x_3,x_3=x_1$。只需赋值使得 $x_1=x_2=x_3$，即可同时满足所有的约束条件。
在第二个问题中，约束条件有四个：$x_1=x_2,x_2=x_3,x_3=x_4,x_4\ne x_1$。由前三个约束条件可以推出 $x_1=x_2=x_3=x_4$，然而最后一个约束条件却要求 $x_1\ne x_4$，因此不可被满足。

【数据范围】

注：实际上 $n\le 10^6$ 。

题意

有$t$个问题，每个问题给出$n$条约束条件，其格式为$x$ $y$ $e$，分两种情况

• $x$ $y$ $1$ $\to$ $a_x=a_y$
• $x$ $y$ $0$ $\to$ $a_x\ne a_y$

求这$n$条约束条件是否可以互相满足 $(YES,NO)$
example：
$a_1=a_2,a_2=a_3,a_1\ne a_3$是不可以互相满足的
$a_1=a_2,a_1=a_3,a_2=a_3$是可以互相满足的

思路

对于$a_x=a_y$，我们可以让 $a_y$ 的父亲指向 $a_x$
当我们判断$a_x$是否等于$a_y$时，就可以$find$两个变量最老的“祖先”，判断二者是不是同一个家族的成员（同一个量）
好像已经完成了，但又好像没有……

---

是的，没有完成
我们回到题目看一下数据大小

$WOW!$
$10^9$大小的编号，直接用并查集的父亲数组储存肯定会$MLE$
可是$n$只有$10^6$，这就说明了父亲数组里有$10^9-10^6$个空内存，纯属浪费
所以我们只要将编号都压缩排序后再储存进父亲数组就可以了
那怎么压缩编号呢？很简单，离散化呀
比如说有这样一组问题

3
1 999999999 1
1 888888888 0
888888888 999999999 1
//压缩前^^^
3
1 3 1
1 2 0
2 3 1
//压缩后^^^

如果不用离散化就要开$10^9$的父亲数组
可是用了离散化压缩编号后就成了非常小的编号了，就可以不用开大数组了
这就结束了吗？

---

不，还没有
还有一个重要的细节
我们在对压缩后的约束条件进行并查集处理前
需要将每一个约束条件的$e$进行排序，将$e$为$1$的排在前面，$e$为$0$的排在后面
为什么呢？
因为所有的约束条件没有先后之分
换句话来说，就是先将相等的条件$(e==1)$用并查集连接起来，然后用不相等的条件$(e==0)$进行判断当前的条件是否成立
其实就是将相等的条件看成已知，将不相等的条件看成要判断的条件

---

离散化细节见代码注释

#include<bits/stdc++.h>
using namespace std;
#define int long long
int t,n,cnt,len;
bool flag;
struct Node{
	int x,y,e;
}node[1000010];
int rak[2000020],dx[1000010],dy[1000010],fa[2000010];
int find(int x){//并查集 find"父亲" 
	if(fa[x]!=x)fa[x]=find(fa[x]);
	return fa[x];
}
void merge(int x,int y){//并查集连接 
	fa[find(x)]=find(y);
	return;
}
bool cmp(Node a,Node b){//对e进行排序 
	return a.e>b.e;
}
signed main(){
	scanf("%lld",&t);
	while(t--){
		scanf("%lld",&n);
		flag=true,cnt=0;//初始化 
		for(int i=1;i<=n;i++)scanf("%lld%lld%lld",&node[i].x,&node[i].y,&node[i].e),rak[++cnt]=node[i].x,rak[++cnt]=node[i].y;
		//初始化rak数组，rak数组的作用是排序去重，方便二分查找元素压缩后对应的编号 
		sort(rak+1,rak+1+cnt);//先排序，因为unique（去重）函数需要对有序数组才能起作用 
		len=unique(rak+1,rak+1+cnt)-(rak+1);//去重，得到去重后的数组长度 
		for(int i=1;i<=len;i++)fa[i]=i;//初始化父亲数组 
		for(int i=1;i<=n;i++)node[i].x=lower_bound(rak+1,rak+1+len,node[i].x)-rak,node[i].y=lower_bound(rak+1,rak+1+len,node[i].y)-rak;
		//用二分查找元素压缩的对应编号，放入结构体，方便对e排序 
		sort(node+1,node+1+n,cmp);//对e进行排序 
		for(int i=1;i<=n;i++){
			if(node[i].e)merge(node[i].x,node[i].y);//若当前约束条件为 x等于y，连接x和y 
			else if(find(node[i].x)==find(node[i].y)){
				//若当前约束条件为 x不等于y，但是在前面的约束条件中 直接 或 间接 表达了 x等于y
				//推翻了当前约束条件，则当前问题的约束条件不可以互相满足 
				flag=false;
				break;
			}
		}
		if(flag)printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}