AcWing 237. 程序自动分析（并查集，离散化）

一.题目

二. 分析

1. 并查集思路

由题意可知，我们赋予x之间的关系，为相等和不等，并且去判断是否出现矛盾，我们发现，赋予关系的顺序与答案是无关的，且当我们只赋予相等关系时，只会将x之间分为若干个集合，不会出现矛盾情况，因此我们可以先只考虑x之间的相等关系，然后用并查集建立集合，再去考虑不等关系，如果xi与xj在同一个集合内，却又存在不等关系，则出现矛盾。

2. 离散化思路

由数据范围可知 i，j 的范围最大可到1e9，我们当然不可能开一个1e9的数组来维护并查集，但我们发现，n的范围只有1e5，最多只会赋予n次关系，即最多出现2e5个不同的数字，且不同数字之间的大小关系对题目无关，因此可以用哈希的离散化思路，利用map将每个出现的数字按照出现顺序，从1开始依次映射，进行离散化。

三. 代码

#include <iostream>
#include <algorithm>
#include <cstring>
#include <unordered_map>
using namespace std;
const int N = 1e5+10;
int p[N*2],n;
unordered_map<int,int> S;
struct Query
{
    int a,b,e;
}query[N];
int get(int x)
{
    if(S.count(x)==0) S[x]= ++n;
    return S[x];
}
int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main()
{
    int t;cin>>t;

    while(t--)
    {
        S.clear();
        n=0;
        int m;cin>>m;
        for(int i=0;i<m;i++)
        {
            int a,b,e;
            cin>>a>>b>>e;
            query[i]={get(a),get(b),e};
        }
        for(int i=1;i<=n;i++) p[i]=i;
        for(int i=0;i<m;i++)
          if(query[i].e==1)
          {
              int pa=find(query[i].a),pb=find(query[i].b);

                  p[pa]=pb;

          }
        bool st=true;
        for(int i=0;i<m;i++)
          if(query[i].e==0)
          {
              int pa=find(query[i].a),pb=find(query[i].b);
              if(pa==pb)
              {
                  st=false;
                  break;
              }
          }
        if(st) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
}