学习C++从娃娃抓起!记录下洛谷C++学习和备考过程中的题目,记录每一个瞬间。
附上汇总贴:洛谷刷题C++语言 | 汇总
【题目描述】
在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本:假设 x 1 , x 2 , x 3 , … x_1,x_2,x_3,\dots x1,x2,x3,… 代表程序中出现的变量,给定 n 个形如 x i = x j x_i=x_j xi=xj 或 x i ≠ x j x_i\neq x_j xi=xj 的变量相等/不等的约束条件,请判定是否可以分别为每一个变量赋予恰当的值,使得上述所有约束条件同时被满足。例如,一个问题中的约束条件为: x 1 = x 2 , x 2 = x 3 , x 3 = x 4 , x 4 ≠ x 1 x_1=x_2,x_2=x_3,x_3=x_4,x_4\neq x_1 x1=x2,x2=x3,x3=x4,x4=x1,这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。
现在给出一些约束满足问题,请分别对它们进行判定。
【输入】
输入的第一行包含一个正整数 t,表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题,包含若干行:
第一行包含一个正整数 n,表示该问题中需要被满足的约束条件个数。接下来 n 行,每行包括三个整数 i,j,e,描述一个相等/不等的约束条件,相邻整数之间用单个空格隔开。若 e=1,则该约束条件为 x_i=x_j。若e=0,则该约束条件为 x i ≠ x j x_i\neq x_j 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
【代码详解】
#include <bits/stdc++.h>
using namespace std;
const int N = 100005;
int t, n, cnt;
int a[N], b[N], e[N], p[N*2];
unordered_map<int, int> mp;
int find(int x)
{
return p[x]==x ? x : p[x] = find(p[x]);
}
int get(int x) //离散化
{
if (mp.count(x)==0) mp[x] = ++cnt;
return mp[x];
}
int main()
{
cin >> t;
while (t--) {
cnt = 0;
mp.clear(); //mp初始化
cin >> n;
for (int i=1; i<=n*2; i++) p[i] = i;
for (int i=1; i<=n; i++) {
cin >> a[i] >> b[i] >> e[i];
a[i] = get(a[i]);
b[i] = get(b[i]);
}
for (int i=1; i<=n; i++) { //合并集合
if (e[i]==1) {
int x = find(a[i]), y = find(b[i]);
if (x!=y) p[x] = y;
}
}
bool flag = 1;
for (int i=1; i<=n; i++) { //判断矛盾
if (e[i]==0 && find(a[i])==find(b[i])) {
flag = 0;
break;
}
}
flag ? puts("YES") : puts("NO");
}
return 0;
}
【运行结果】
2
2
1 2 1
1 2 0
NO
2
1 2 1
2 1 1
YES