[NOI2015]程序自动分析 --- 并查集

传送门：洛谷P1955

---

题目描述

在实现程序自动分析的过程中，常常需要判定一些约束条件是否能被同时满足。

考虑一个约束满足问题的简化版本：假设x1,x2,x3…代表程序中出现的变量，给定n个形如xi=xj或xi≠xj的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。例如，一个问题中的约束条件为：x1=x2,x2=x3,x3=x4,x4≠x1，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

现在给出一些约束满足问题，请分别对它们进行判定。

---

分析

　　其他的不说，看到数据范围就知道要走一波离散化。
　　本来想着用带权并查集做的（用没加离散化的带权并查集莫名水了70分），但由于不等之间没有确定的关系，所以有点麻烦，就放弃了。
　　但想想之后倒也简单，可以考虑先连边后判断，即相等的先全部连边，再用不相等的做判断

---

代码

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>

#define IL inline
#define open(s) freopen(s".in","r",stdin); freopen(s".out","w",stdout);
#define close fclose(stdin); fclose(stdout);

using namespace std;

IL int read()
{
    char c = getchar();
    int sum = 0 ,k = 1;
    for(;'0' > c || c > '9'; c = getchar())
        if(c == '-') k = -1;
    for(;'0' <= c && c <= '9'; c = getchar()) sum = sum * 10 + c - '0';
    return sum * k;
}

struct Question
{
    int x, y;
    IL Question(int x_ = 0, int y_ = 0)
    {
        x = x_; y = y_;
    }
}q[2][100000 + 5];
int tq[2];

int n, m;
bool ans;
int num[200000 + 5];
int fa[200000 + 5];
int size[200000 + 5];

IL void clean()
{
    m = tq[0] = tq[1] = 0; ans = 1;
}

IL int find(int x)
{
    return fa[x] = (fa[x] == x ? x : find(fa[x]));
}

IL void merge(int x, int y)//启发式合并
{
    int x1 = find(x), y1 = find(y);
    if(x1 == y1) return ;
    if(size[x1] > size[y1])
    {
        size[x1] += size[y1];
        fa[y1] = x1;
    }else
    {
        size[y1] += size[x1];
        fa[x1] = y1;
    }
}

int main()
{
    open("1955")

    for(int T = read(); T; --T)
    {
        clean();

        n = read();
        for(int i = 1, x, y, z; i <= n; ++i)
        {
            x = read(); y = read(); z = read();
            num[++m] = x; num[++m] = y;
            q[z][++tq[z]] = Question(x, y);
        }

        sort(num + 1, num + m + 1);
        m = unique(num + 1, num + m + 1) - (num + 1);

        for(int i = 1; i <= m; ++i) {fa[i] = i; size[i] = 1;}

        for(int k = 1; k >= 0; --k)
        for(int i = 1, x, y; i <= tq[k]; ++i)
        {
            x = lower_bound(num + 1, num + m + 1, q[k][i].x) - num; 
            y = lower_bound(num + 1, num + m + 1, q[k][i].y) - num; 
            if(k)merge(x, y); else
            if(ans && find(x) == find(y)) ans = 0;
        }
        printf("%s\n", ans ? "YES" : "NO");
    }

    close
    return 0;
}