代码源：952、程序自动分析

题目描述

这是代码源5月18日的div2每日一题

题面

最喜欢数学了（高数除外） - 题目 - Daimayuan Online Judge

小明的数学很好，所以他认为世界上所有的数学问题都很简单。

但是有一天，他遇到了一个他无法解决的数学问题，所以他请求你帮助他。

小明会给你两个数字 a 和 b ，然后你应该选择一个正奇数 x 和一个正偶数 y ，你可以让 a 加上 x 或让 a 减去 y，在一次运算中。你应该在最少的操作中把 a 变成 b。请注意，在同一次运算中，你不允许改变 x 和 y 的值。

输入格式

在第一行，有一个整数 T。

接下来 T 行每行2个整数 a、b。表示由小明给出的数字。

输出格式

一个数，表示将 a 变为 b 所需的最小操作数。。

样例输入

1
10 7

样例输出

2

数据规模

所有数据保证 1≤T≤10^5,1≤a,b≤10^6。

问题解析

并查集模板题。

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

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

AC代码

#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;
}