P1955 程序自动分析

题面:https://www.luogu.org/problemnew/show/P1955

本题可以把等于的并在一起,碰到不等于的就判断是否矛盾。由于数据比较大，注意要离散化。

Code:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<ctime>
using namespace std;
const int N=1000005;
int t,n,len,b[N],fa[N];
bool flag;
struct Node{
    int i,j,e;
}a[N];
bool cmp(Node a,Node b){
    return a.e>b.e;
}
int find(int x){
    if(fa[x]==x){
        return fa[x];
    }
    return fa[x]=find(fa[x]);
}
void merge(int a,int b){
    int x=find(a),y=find(b);
    if(x!=y){
        fa[x]=y;
    }
}
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        len=0;
        for(int i=1;i<=n;i++){
            scanf("%d%d%d",&a[i].i,&a[i].j,&a[i].e);
            b[++len]=a[i].i;
            b[++len]=a[i].j;
        }
        sort(a+1,a+1+n,cmp);
        sort(b+1,b+1+len);
        len=unique(b+1,b+len+1)-(b+1);
        for(int i=1;i<=len;i++){
            fa[i]=i;
        }
        flag=true;
        for(int i=1;i<=n;i++){
            a[i].i=lower_bound(b+1,b+len+1,a[i].i)-b;
            a[i].j=lower_bound(b+1,b+len+1,a[i].j)-b;
            if(a[i].e){
                merge(a[i].i,a[i].j);
            }
            else if(find(a[i].i)==find(a[i].j)){
                flag=false;
                break;
            }
        }
        if(flag){
            printf("YES\n");
        }
        else{
            printf("NO\n");
        }
    }
    return 0;
}