程序自动分析
思路:一个简单的并查集问题,不过需要离散化把大的数值映射成小的,先把相等的放在一个集合,最后在判断不相等的是否与相等的冲突
具体代码如下
#include<iostream>
#include<unordered_map>
using namespace std;
const int N = 2000010;
int n, m;
int p[N];
unordered_map<int, int> S; //这里用hash表来映射就可以,也可以自己手写一个hash表
struct Q{
int x, y, e;
}query[N];
int get(int x){
if(!S.count(x)) S[x] = ++n; //如果x还没有就把x映射到没有映射过的数值上
return S[x];
}
int find(int x){
if(p[x] != x) p[x] = find(p[x]);
return p[x];
}
int main(){
int T;
scanf("%d", &T);
while(T--){
n = 0;
S.clear();
scanf("%d", &m);
for(int i=0; i<m; ++i){
int x, y, e;
scanf("%d%d%d", &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 == 1){
int px = find(query[i].x), py = find(query[i].y);
p[px] = py;
}
bool flag = false;
for(int i=0; i<m; ++i)
if(!query[i].e){
int px = find(query[i].x), py = find(query[i].y);
if(px == py){
flag = true;
break;
}
}
if(flag) puts("NO");
else puts("YES");
}
return 0;
}