洛谷P1995 程序自动分析

2017-03-18

题目：https://www.luogu.org/problem/show?pid=1955

这道题一般能看出来是并查集。

但是，i、j的范围似乎有一点鬼畜。。。。。。

但是n的范围还是比较友好的。

所以，我们要用的一招叫做离散化。

离散化，针对只在乎大小，而不在乎具体值的数据使用。

如果不会离散化的话，问度娘或点我>_<。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;

struct node{
    int x,y;
    int r;
}a[100010];

int n,t,c[200210],tot,fa[200210];
int size;
bool yes=true;

int find(int x){
    if (fa[x]!=x) fa[x]=find(fa[x]);
    return fa[x];
}

int main(){
    ios::sync_with_stdio(false);
    cin>>t;
    while (t--){
        yes=true;
        cin>>n;
        tot=0;
        memset(c,0,sizeof(c));
        for (int i=1;i<=2*n;++i) fa[i]=i;
        for (int i=1;i<=n;++i){
            cin>>a[i].x>>a[i].y>>a[i].r;
            c[++tot]=a[i].x;
            c[++tot]=a[i].y;
        }
        sort(c+1,c+1+tot);
        size=unique(c+1,c+1+tot)-c-1;
        for (int i=1;i<=n;++i){
            a[i].x=upper_bound(c+1,c+1+size,a[i].x)-c-1;
            a[i].y=upper_bound(c+1,c+1+size,a[i].y)-c-1;
        }
        for (int i=1;i<=n;++i){
            if (a[i].r){
                int f1=find(a[i].x),f2=find(a[i].y);
                if (f1!=f2) fa[f1]=f2;
            }
        }
        for (int i=1;i<=n;++i){
            if (!a[i].r){
                int f1=find(a[i].x),f2=find(a[i].y);
                if (f1==f2){
                    yes=false;
                    break;
                }
            }
        }
        if (yes) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}

骗分真神奇，暴力出奇迹。