地址 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;
}
标签:约束条件,NoEqualV,进阶,get,int,mm,第五章,237,find
From: https://www.cnblogs.com/itdef/p/17337134.html