牛客小白月赛24 H 人人都是好朋友

人人都是好朋友

时间限制：C/C++ 2秒，其他语言4秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述:
牛可乐作为三军统帅，是要时时刻刻关照着下属的。
现在牛可乐想要知道自己的手下之间的友好关系，所以他收集了 nn 张纸条，上面写着三个整数 ai,bi,ci，表示如果 ci为 1，表示手下 ai和手下 bi是朋友，反之则是敌人。
牛可乐想要知道这些信息有没有互相矛盾的地方，可是这个问题太难了，只好来问你了
如果 A 与 B 友好，B 又与 C 友好，那么 A 与 C 也是友好的。
如果两个人既是友好的又是不友好的则视为相互矛盾的。
牛可乐的手下有 1e9 个。

输入描述:
输入第一行给出一个正整数T,表示测试案例的数量。
接下来每个测试用例，第一行给出一个正整数n,表示有n个友好关系
接下来每n行给出3个正整数ai,bi,ci,表示手下ai和手下bi之间的友好关系

输出描述:
每组案例输出一行，若这些关系没有矛盾，输出 "YES”，否则输出 “NO”

示例1

输入
2
3
1 2 1
1 3 1
2 3 1
3
1 2 1
1 3 1
2 3 0

输出
YES
NO

备注:
1<=T<=10
1<=n<=1e6
1<=a,b<=1e9
c∈{0,1}
对于每组样例,保证∑n<=1010000

解题思路:
这题数据范围很大,一般的并查集没有这么大的内存空间来保存数据,离散化也不会写,就借助stl库中的map作为数组使用.先把ci为1的所有关系并在一起,然后遍历ci为0的每队关系,查看是否存在冲突即可.

代码如下:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6+5;
int n,a[N],b[N],c[N];
unordered_map<int,int>bin;
inline int find(int a){
    if(bin.find(a)==bin.end()||bin[a]==a)return bin[a]=a;
    return bin[a]=find(bin[a]);
}
inline void merge(int a,int b){
    int fa=find(a);
    int fb=find(b);
    if(fa!=fb){
        bin[fa]=fb;
    }
}
int main(){
    ios::sync_with_stdio(false);
    int t;
    cin>>t;
    while(t--){
        bin.clear();//别忘了初始化
        cin>>n;
        int ans=0;
        for(int i=0;i<n;i++){
            cin>>a[ans]>>b[ans]>>c[ans];
            if(c[ans]==1){
                merge(a[ans],b[ans]);
            }else{
                ans++;
            }
        }
        bool flag=true;
        for(int i=0;i<ans;i++){
            if(find(a[i])==find(b[i])){
                flag=false;
                break;
            }
        }
        if(flag)cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}