bzoj4195: [Noi2015]程序自动分析（离散化+并查集）

题目传送门
。

解法：
相等的可以用一个并查集维护在一起。
那么当所有相等的都连好以后。
问所有不相等的。如果不相等的在同一个块那么说明有问题。。
离散化是显然的啦

代码实现：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
struct node {int s,id;}a[2100000],b[2100000];
struct trnode {int x,y,next;}e[2100000];int len,last[1100000];
void ins(int x,int y) {len++;e[len].x=x;e[len].y=y;e[len].next=last[x];last[x]=len;}
bool cmp(node n1,node n2) {return n1.s<n2.s;}int v[1100000];bool bo[1100000];
int fa[1100000];int findfa(int x) {if(x!=fa[x])fa[x]=findfa(fa[x]);return fa[x];}
int main() {
    int T;scanf("%d",&T);
    while(T--) {
        int n;scanf("%d",&n);int t=0;
        for(int i=1;i<=n;i++) {
            t++;scanf("%d",&a[t].s);a[t].id=i;
            t++;scanf("%d",&a[t].s);a[t].id=i;
            scanf("%d",&v[i]);
        }sort(a+1,a+1+t,cmp);
        int tt=0;memset(bo,false,sizeof(bo));
        for(int i=1;i<=t;i++) {
            if(a[i].s!=a[i-1].s)tt++;
            if(bo[a[i].id]==false) {bo[a[i].id]=true;b[a[i].id*2-1].s=tt;}
            else b[a[i].id*2].s=tt;
        }for(int i=1;i<=tt;i++)fa[i]=i;
        len=0;memset(last,0,sizeof(last));
        for(int i=1;i<=n;i++) {
            int x=b[i*2-1].s,y=b[i*2].s;
            if(v[i]==1) {
                int xx=findfa(x),yy=findfa(y);
                if(xx!=yy)fa[xx]=yy;
            }
        }
        bool bk=true;
        for(int i=1;i<=n;i++) {
            int x=b[i*2-1].s,y=b[i*2].s;
            if(v[i]==0) {
                int xx=findfa(x),yy=findfa(y);
                if(xx==yy){bk=false;break;}
            }
        }
        if(bk==false)printf("NO\n");
        else printf("YES\n");
    }
    return 0;
}