程序自动分析【并查集】

程序自动分析

题目

思路：一个简单的并查集问题，不过需要离散化把大的数值映射成小的，先把相等的放在一个集合，最后在判断不相等的是否与相等的冲突

具体代码如下

#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;
}