《算法竞赛进阶指南》 第五章 237. 程序自动分析

发布时间 2023-04-20 15:46:16作者: itdef

地址 https://www.acwing.com/problem/content/239/

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

考虑一个约束满足问题的简化版本:假设 x1,x2,x3,… 代表程序中出现的变量,
给定 n 个形如 xi=xj 或 xi≠xj 的变量相等/不等的约束条件,请判定是否可以分别为每一个变量赋予恰当的值,
使得上述所有约束条件同时被满足。

例如,一个问题中的约束条件为:x1=x2,x2=x3,x3=x4,x1≠x4,
这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。

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

输入格式
输入文件的第 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 表示不可被满足。

数据范围
1≤n≤105
1≤i,j≤109
输入样例:
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
输出样例:
NO
YES

解答
使用并查集即可, 注意不是加权或者扩展域并查集,因为a!=b 和c!=b 的情况不能说明a和c的关系,所以使用边权扩展域这种非a即b否则为c的情况。
同时注意数值较少但是取值范围很大,所以并查集的数组范围不能取这么多,需要离散化

#include <vector>
#include <unordered_map>
#include <iostream>
using namespace std;

const int N = 2000010 +10;
unordered_map<int, int> mm; int midx;
int n,t;
int p[N];
vector<pair<int, int>> NoEqualV;

void initUnion() {
	for (int i = 0; i < N; i++) {
		p[i] = i;
	}
}

int get(int x) {
	if (mm.count(x) == 0) {
		mm[x] = ++midx;
		return midx;
	}
	return mm[x];
}

int find(int x) {
	if (p[x] != x) p[x] = find(p[x]);
	return p[x];
}

int main()
{
	cin >> t;
	while (t--) {
		cin >> n;
		int x, y, c;
		initUnion();
		NoEqualV.clear();
		mm.clear();
		for (int i = 0; i < n; i++) {
			cin >> x >> y >> c;
			int xn = get(x);
			int yn = get(y);
			if (c == 1) {
				//合并所有的相等项
				int fx = find(xn); int fy = find(yn);
				p[fx] = fy;
			}
			else if (c == 0) {
				NoEqualV.push_back({ xn,yn });
			}
		}

		//再次查找不等的式子 查看是否有冲突
		int conflict = 0;
		for (int i = 0; i < NoEqualV.size(); i++) {
			int an = NoEqualV[i].first; int bn = NoEqualV[i].second;
			//int an = get(a); int bn = get(b);
			int fa = find(an); int fb = find(bn);
			if (fa == fb) {
				conflict = 1; break;
			}
		}

		if (1 == conflict) {cout << "NO" << endl;}
		else {	cout << "YES" << endl;}
	}


	return 0;
}