【bzoj 4159】程序自動分析

傳送門~

解題思路

先把所有相等情況加進並查集，在判不等就行了。
代碼：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
#include<cmath>
#include<cstdlib>
using namespace std;
struct ldx{
    int x,y,c;
    bool operator < (const ldx p) const{
        return c>p.c;
    }
}s[100005];
int b[200005];
int fa[200005];
int n,tot,T;
int zh(int x){
    if(x==fa[x]) return x;
    else return fa[x]=zh(fa[x]);
}
int main(){
    scanf("%d",&T);
while(T--){
    scanf("%d",&n);tot=0;
    for(int i=1;i<=n*2;i++) fa[i]=i;
    for(int i=1;i<=n;i++){
        scanf("%d%d%d",&s[i].x,&s[i].y,&s[i].c);
        b[++tot]=s[i].x;b[++tot]=s[i].y;
    }
    sort(b+1,b+tot+1);sort(s+1,s+n+1);
    int top=unique(b+1,b+tot+1)-b-1;
    for(int i=1;i<=n;i++){
        s[i].x=lower_bound(b+1,b+top+1,s[i].x)-b;
        s[i].y=lower_bound(b+1,b+top+1,s[i].y)-b;
    }
    int r1,r2,flag=0;
    for(int i=1;i<=n;i++){
        r1=zh(s[i].x);r2=zh(s[i].y);
        if(s[i].c && r1!=r2) fa[r1]=r2;     
        else if(!s[i].c && r1==r2) {flag=1;break;}
    }
    if(flag) printf("NO\n");
    else printf("YES\n");   
}
    return 0;
}