一.题目
二. 分析
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;
}
}