$\huge \color{orange}{成魔之路->}$ $\huge \color{purple}{算法提高课题解}$
思路:
1. 需要合并的数范围很大,但用到的数很少,故需要进行离散化操作
2. 将所有相等条件的数合并集合
3. 判断剩下所有的不等条件,如果在同一集合就说明存在冲突
完整代码
#include<bits/stdc++.h>
using namespace std;
const int N = 200010;
int n,m;
int p[N];
unordered_map<int,int> S;
struct Query
{
int x,y,e;
}query[N];
//离散化操作
int get(int x)
{
if(!S.count(x)) 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--)
{
//初始化
n=0;
S.clear();
cin>>m;
for(int i=0;i<m;i++)
{
int x,y,e;
cin>>x>>y>>e;
query[i]={get(x),get(y),e};
}
//并查集初始化操组
for(int i=1;i<=n;i++) p[i]=i;
//把相等条件的合并
for(int i=0;i<m;i++)
if(query[i].e)
p[find(query[i].x)]=find(query[i].y);
//判断剩下的不等条件,如果在同一个集合就说明有冲突
bool has_conflict=false;
for(int i=0;i<m;i++)
if(!query[i].e&&find(query[i].x)==find(query[i].y))
{
has_conflict=true;
break;
}
if(has_conflict) cout<<"NO"<<endl;
else cout<<"YES"<<endl;
}
return 0;
}