在刚刚结束的NOI网赛里被虐了,只好写一下T1题解来发泄QAQ
题目大意:给你n个判定式(等式或不等式),让你判定这些式子是否冲突。
对于这道题,我们可以用并查集来保存等式,每枚举一个等式,把相等的合并,后来判断不等式,检测到非法的后就输出NO。
/**************************************************************
Problem: 4195
User: geng4512
Language: C++
Result: Accepted
Time:1416 ms
Memory:32092 kb
****************************************************************/
#include<cstdio>
#include<stack>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 1000010;
int N, T;
int fa[MAXN*2];
inline void GET(int &r) { //n很大,数据组数也多
char c; r=0;
do c=getchar(); while (c<'0'||c>'9');
do{r=r*10+c-'0';c=getchar();}while (c>='0'&&c<='9');
}
int data[MAXN][2]; //记录输入数据以离散化
int b[MAXN*2]; //记录是等号还是不等号
int arr[MAXN * 2], cnt, len;
stack<int> z;
inline int bifind(int x) { //二分查找(lower_bound比手写的慢三倍。。)
int l = 1, r = len, mid = (l + r) >> 1;
while (l < r) {
if (arr[mid] == x) return mid;
if (arr[mid] < x) l = mid + 1;
else r = mid - 1;
mid = (l + r) >> 1;
}
return mid;
}
inline void init() {
for (int i = 1; i<=len; ++i) //这里是len不是n。。n是关系的个数,len才是变量个数
fa[i] = i;
}
inline int root(int x) { //用栈实现的路径压缩防止递归爆栈。然而由于路径压缩的存在并不需要。。
while (fa[x] != x && x != -1) z.push(x), x = fa[x];
while (!z.empty()) fa[z.top()] = x, z.pop();
return x;
}
inline bool solve() {
init();
int i, t1, t2;
for (i = 1; i<=N; ++i) //先将所有相等关系合并成集合
if (b[i] == 1) {
t1 = bifind(data[i][0]);
t2 = bifind(data[i][1]);
if (root(t1) != root(t2))
fa[root(t1)] = root(t2);
}
for (i = 1; i<=N; ++i)
if (b[i] == 0) {
t1 = bifind(data[i][0]);
t2 = bifind(data[i][1]);
if (root(t1) == root(t2)) //检查是否冲突
return 0;
}
return 1;
}
int main()
{
int i;
GET(T);
while (T--) {
GET(N);
cnt = 0;
for (i = 1; i <= N; i ++) {
GET(data[i][0]); GET(data[i][1]);
GET(b[i]);
arr[++cnt] = data[i][0];
arr[++cnt] = data[i][1];
}
sort(arr + 1, arr + cnt + 1);
len = unique(arr + 1, arr + cnt + 1) - (arr + 1); //len保存离散化后变量个数(重复出现不计)
if (solve()) puts("YES");
else puts("NO");
}
return 0;
}