P1955 [NOI2015]程序自动分析

题目描述

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

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

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

输入格式

输入的第一行包含一个正整数 tt，表示需要判定的问题个数。注意这些问题之间是相互独立的。

对于每个问题，包含若干行：

第一行包含一个正整数 nn，表示该问题中需要被满足的约束条件个数。接下来 nn 行，每行包括三个整数 i,j,ei,j,e，描述一个相等/不等的约束条件，相邻整数之间用单个空格隔开。若 e=1e=1，则该约束条件为 x_i=x_jxi​=xj​。若e=0e=0，则该约束条件为 x_i\neq x_jxi​​=xj​。

输出格式

输出包括 tt 行。

输出文件的第 kk 行输出一个字符串 YES 或者 NO（字母全部大写），YES 表示输入中的第 kk 个问题判定为可以被满足，NO 表示不可被满足。

输入输出样例

输入 #1复制

2
2
1 2 1
1 2 0
2
1 2 1
2 1 1

输出 #1复制

NO
YES

输入 #2复制

2
3
1 2 1
2 3 1
3 1 1
4
1 2 1
2 3 1
3 4 1
1 4 0

输出 #2复制

YES
NO

说明/提示

【样例解释1】

在第一个问题中，约束条件为：x_1=x_2,x_1\neq x_2x1​=x2​,x1​​=x2​。这两个约束条件互相矛盾，因此不可被同时满足。

在第二个问题中，约束条件为：x_1=x_2,x_1 = x_2x1​=x2​,x1​=x2​。这两个约束条件是等价的，可以被同时满足。

【样例说明2】

在第一个问题中，约束条件有三个：x_1=x_2,x_2= x_3,x_3=x_1x1​=x2​,x2​=x3​,x3​=x1​。只需赋值使得 x_1=x_2=x_3x1​=x2​=x3​，即可同时满足所有的约束条件。

在第二个问题中，约束条件有四个：x_1=x_2,x_2= x_3,x_3=x_4,x_4\neq x_1x1​=x2​,x2​=x3​,x3​=x4​,x4​​=x1​。由前三个约束条件可以推出 x_1=x_2=x_3=x_4x1​=x2​=x3​=x4​，然而最后一个约束条件却要求 x_1\neq x_4x1​​=x4​，因此不可被满足。

【数据范围】

注：实际上 n\le 10^6n≤106 。

思路：对于相等的情况，把他们放到一个集合里，对于不等的情况若他们已经在一个集合里了，矛盾就输出no。先排序将e为1的放前面，e为0的放后面。

并查集需要开个数组，由题中数据可知，数组开不了那么大，所以想到离散化1 ，离散化2

#include<cstdio>
#include<iostream>
#include<vector>
#include<iomanip>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<algorithm>
#include<string>
#include<cmath>
#include<iterator>
using namespace std;


struct Node {
    int i, j;
    int e;
};

Node r[1000000];
int t, n, i, j, e;
int f[1000000];



bool comp(Node a, Node b) {
    return a.e > b.e;
}




int find(int a) {
    if (f[a] == a)return a;
    else return f[a] = find(f[a]);
}

void merge(int a, int b) {
    int ta = find(a);
    int tb = find(b);
    if (ta != tb)f[tb] = ta;
}

void init() {
    for (int i = 1; i <= 1000000; i++) {
        f[i] = i;
    }
}


int main()
{
    cin >> t;
    for (int i = 0; i < t; i++) {
        vector<int> p;
        init();
        cin >> n;
        for (int j = 0; j < n; j++) {
            scanf("%d %d %d",&r[j].i,&r[j].j,&r[j].e);
            p.push_back(r[j].i);
            p.push_back(r[j].j);
        }
        
        
        sort(r, r + n, comp);
        //离散化 1.排序
        sort(p.begin(), p.end());
        //2.去重
        p.erase(unique(p.begin(), p.end()), p.end());
        int f = 0;
        for (int i = 0; i < n; i++) {
            //3.索引
            r[i].i = lower_bound(p.begin(), p.end(), r[i].i) - p.begin() + 1;
            r[i].j = lower_bound(p.begin(), p.end(), r[i].j) - p.begin() + 1;
            if (r[i].e == 1) {
                merge(r[i].i, r[i].j);
            }
            else {
                if (find(r[i].j) == find(r[i].i)) {
                    cout << "NO" << endl;
                    f = 1;
                    break;
                }
            }
        }
        if(!f)
            cout << "YES" << endl;
    }

    
    return 0;
}