洛谷-程序自动分析

题目描述：

输入格式：

输出格式：

输入输出样例：
输入样例1：

2
2
1 2 1
1 2 0
2
1 2 1
2 1 1

输出样例1：

NO
YES

输入样例2：

2
3
1 2 1
2 3 1
3 1 1
4
1 2 1
2 3 1
3 4 1
1 4 0

输出样例2：

YES
NO

思路：
这个题是典型的并查集的题目，我们先处理所有的e=1的情况，将所有的相等的情况并在一起，再处理e=0的时候，当我们出现矛盾（即我们已经将某两个数字并在了一起，但是这两个是不能相等的）时，我们便输出NO，如果整个过程都没有输出NO，则输出YES。
难点：
由于这个题目本身的范围极大，为10⁹大小，加入按照我们常规做法，我们会开一个10⁹大小的数组，这样做会出现MLE，RE等错误提示。 因此我们需要用到离散化的思想做法，将数组排好序，然后利用stl的unique得到我们的实际大小的数组。再进行我们的常规的并查集操作。
技巧：离散化过程
AC代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <cstdlib>
using namespace std;
int t,n,f[1000007],book[1000007*3];  //t表示t组数据，n表示有n个操作，f[]是我们并查集的数字，book[]是离散化的数组 
struct node{
    int x,y,e;
}a[1000001];  
bool cmp(node a,node b){
    return a.e>b.e;
}//排 序将e==1的放在前面 
inline void first(int kkk){
    for(int i=1;i<=kkk;i++)  f[i]=i;
}//初 始 化 
int get(int x){
    if(x==f[x]) return x;
    return f[x]=get(f[x]);
}
int main(){
    scanf("%d",&t);
    while(t--){
      int tot=-1;
      memset(book,0,sizeof(book));
      memset(a,0,sizeof(a));
      memset(f,0,sizeof(f));
    int flag=1;
        scanf("%d",&n);
       
        for(int i=1;i<=n;i++){
            scanf("%d %d %d",&a[i].x,&a[i].y,&a[i].e);
            book[++tot]=a[i].x;
            book[++tot]=a[i].y;
        }
        sort(book,book+tot);//排序 
        int reu=unique(book,book+tot)-book;  //去重 
        for(int i=1;i<=n;++i){
           a[i].x=lower_bound(book,book+reu,a[i].x)-book;
           a[i].y=lower_bound(book,book+reu,a[i].y)-book;   
        } 
        first(reu);   //初始化 
        sort(a+1,a+n+1,cmp);  //按e排序 
        for(int i=1;i<=n;i++){
            int r1=get(a[i].x);
            int r2=get(a[i].y);
            if(a[i].e){
                f[r1]=r2;  //就是我们的merge操作 
            }else if(r1==r2){
                printf("NO\n");
                flag=0;  //如果不满足条件，标记为否 
                break;
            }
        }
        if(flag)    printf("YES\n");   //都满足条件了 
    }
    return 0;
}