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;
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值