NOI2015 Day1 T1 程序自动分析

在刚刚结束的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;
}