[NOI2015]程序自动分析

OJ题号：

BZOJ4195、洛谷1955

思路：

首先将信息的下标离散化，首先处理$e=1$时的情况，直接用并查集合并。
对于$e=0$时的情况，判断两个是否在统一集合。

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#include<functional>
#include<ext/hash_map>
inline int getint() {
    char ch;
    while(!isdigit(ch=getchar()));
    int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
__gnu_cxx::hash_map<int,int> id;
int sz;
int hash(const int x) {
    return id[x]?id[x]:id[x]=++sz;
}
class DisjointSet {
    private:
        int anc[200001];
        int Find(const int x) {
            return x==anc[x]?x:anc[x]=Find(anc[x]);
        }
    public:
        void reset() {
            for(int i=0;i<=200000;i++) anc[i]=i;
        }
        bool isConnected(const int x,const int y) {
            return Find(x)==Find(y);
        }
        void Union(const int x,const int y) {
            anc[Find(x)]=Find(y);
        }
};
DisjointSet s;
bool flag;
void reset() {
    id.clear();
    sz=0;
    flag=false;
    s.reset();
}
struct Query {
    int i,j,e;
    bool operator > (const Query &x) const {
        return e>x.e;
    }
};
Query q[100000];
int main() {
    for(int T=getint();T;T--) {
        reset();
        int n=getint();
        for(int i=0;i<n;i++) {
            q[i].i=hash(getint()),q[i].j=hash(getint()),q[i].e=getint();
        }
        std::sort(&q[0],&q[n],std::greater<Query>());
        for(int i=0;i<n;i++) {
            if(q[i].e) {
                s.Union(q[i].i,q[i].j);
            }
            else {
                if(s.isConnected(q[i].i,q[i].j)) {
                    puts("NO");
                    flag=true;
                    break;
                }
            }
        }
        if(!flag) puts("YES");
    }
    return 0;
}