2022-05-18每日刷题打卡

2022-05-18每日刷题打卡

题面

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

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

例如，一个问题中的约束条件为：x1=x2,x2=x3,x3=x4,x1≠x4 这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

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

输入格式

输入文件的第一行包含一个正整数 t (1≤t≤105)，表示需要判定的问题个数，注意这些问题之间是相互独立的。

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

第一行包含一个正整数 n 1≤n≤105)，表示该问题中需要被满足的约束条件个数。

接下来 n 行，每行包括 3 个整数 i,j,e (1≤i,j≤109,0≤e≤1)，描述 1 个相等/不等的约束条件，相邻整数之间用单个空格隔开。若 e=1，则该约束条件为 xi=xj；若 e=0，则该约束条件为 xi≠xj 。

输出格式

输出文件包括 t 行。

输出文件的第 k 行输出一个字符串 YES 或者 NO，YES 表示输入中的第 k 个问题判定为可以被满足，NO 表示不可被满足。

输入样例

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

输出样例

NO
YES

问题解析

并查集模板题。

我们可以先把所有的相等操作拿出来先处理，把相同的数都形成一个连通块，在同一个连通块里的数就代表他们之间两两相同，然后我们再处理所有不相等的操作，如果两个不相同的数出现在同一个连通块里，就说明不可被满足，直接输出no，要是跑完全部不相等操作也没输出no，就输出yes。

（注意题目数据元素最大可以到1e9，所以要先进行离散化操作）

#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>
#include<math.h>
#include<set>
#include<numeric>
#include<string>
#include<string.h>
#include<iterator>
#include<map>
#include<unordered_map>
#include<stack>
#include<list>
#include<queue>
#include<iomanip>

#define endl '\n';
typedef long long ll;
typedef pair<ll, ll>PII;
const int N = 1e6 + 50;
int f[N],mysize[N];

int find(int x)
{
    if (x != f[x])
        f[x] = find(f[x]);
    return f[x];
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int t;
    cin >> t;
    while (t--)
    {
        ll n, x, y, e;
        cin >> n;
        vector<int>v;
        vector<PII>ans[2];
        unordered_map<int, int>mymap;
        while (n--)
        {
            cin >> x >> y >> e;
            ans[e].push_back({ x,y });
            v.push_back(x);
            v.push_back(y);
        }
        sort(v.begin(), v.end());
        v.erase(unique(v.begin(), v.end()), v.end());
        for (int i = 0; i < v.size(); i++)
            mymap[v[i]] = i + 1, f[i + 1] = i + 1, mysize[i + 1] = 1;
        for (auto i : ans[1])
        {
            int a = i.first, b = i.second;
            int x = find(mymap[a]), y = find(mymap[b]);
            if (x != y)
            {
                if (mysize[x] > mysize[y])swap(x, y);
                mysize[y] += mysize[x];
                f[x] = y;
            }
        }
        bool flag = true;
        for (auto i : ans[0])
        {
            int a = i.first, b = i.second;
            int x = find(mymap[a]), y = find(mymap[b]);
            if (x == y)
            {
                flag = false;
                break;
            }
        }
        if (flag)cout << "YES";
        else cout << "NO";
        cout << endl;
    }
    return 0;
}