BZOJ 4195: [Noi2015]程序自动分析
Description
在实现程序自动分析的过程中,常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本:假设x1,x2,x3,…代表程序中出现的变量,给定n个形如xi=xj或xi≠xj的变量相等/不等的约束条件,请判定是否可以分别为每一个变量赋予恰当的值,使得上述所有约束条件同时被满足。例如,一个问题中的约束条件为:x1=x2,x2=x3,x3=x4,x1≠x4,这些约束条件显然是不可能同时被满足的,因此这个问题应判定为不可被满足。
现在给出一些约束满足问题,请分别对它们进行判定。
Input
输入文件的第1行包含1个正整数t,表示需要判定的问题个数。注意这些问题之间是相互独立的。
对于每个问题,包含若干行:
第1行包含1个正整数n,表示该问题中需要被满足的约束条件个数。
接下来n行,每行包括3个整数i,j,e,描述1个相等/不等的约束条件,相邻整数之间用单个空格隔开。若e=1,则该约束条件为xi=xj;若e=0,则该约束条件为xi≠xj。
Output
输出文件包括t行。
输出文件的第k行输出一个字符串“YES”或者“NO”(不包含引号,字母全部大写),“YES”表示输入中的第k个问题判定为可以被满足,“NO”表示不可被满足。
Sample Input
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
Sample Output
NO
YES
HINT
在第一个问题中,约束条件为:x1=x2,x1≠x2。这两个约束条件互相矛盾,因此不可被同时满足。
在第二个问题中,约束条件为:x1=x2,x2=x1。这两个约束条件是等价的,可以被同时满足。
1≤n≤1000000
1≤i,j≤1000000000
并查集,先对相等的情况建立并查集,最后判断不等的情况,如果两个x不等而且有公共祖先那么就不符合条件
i,j的范围很大,可以先离散化,把范围映射到1-2n内
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
static const auto io_sync_off = []() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
return nullptr;
}();
const int maxn = 1000005;
int N, cnt, tot;
int arr[maxn * 2], num[maxn * 2];// 保存所有的情况,离散化后结果
int f[maxn * 2];// 父亲节点
struct point
{
int x, y, e;
bool operator<(const point &rhs) const// 先处理相等的情况
{
return e > rhs.e;
}
} point[maxn];
void discrete()// 离散化
{
sort(arr + 1, arr + cnt + 1);
for (int i = 1; i <= cnt; ++i)
if (i == 1 || arr[i] != arr[i - 1])
num[++tot] = arr[i];
}
int query(int x)
{
return lower_bound(num + 1, num + tot + 1, x) - num;
}
void init()
{
for (int i = 1; i <= maxn; ++i)
f[i] = i;
}
int find(int x)
{
return f[x] == x ? x : f[x] = find(f[x]);
}
int main()
{
int T;
cin >> T;
while (T--)
{
// 注意初始化
init();
bool flag = 0;
cnt = 0, tot = 0;
cin >> N;
for (int i = 1; i <= N; ++i)
{
cin >> point[i].x >> point[i].y >> point[i].e;
arr[++cnt] = point[i].x;
arr[++cnt] = point[i].y;
}
sort(point + 1, point + N + 1);//先处理相等的情况
discrete();
for (int i = 1; i <= N; ++i)
{
int x = query(point[i].x), y = query(point[i].y);
int fx = find(x), fy = find(y);
if (point[i].e)
f[fy] = fx;
else if (fx == fy)
{
flag = 1;
break;
}
}
if (flag)
cout << "NO" << endl;
else
cout << "YES" << endl;
}
return 0;
}