P1955【绿】

发布时间 2023-11-01 14:33:05作者: Kai-G

这道题是标准的“离散化+并查集”模版题,通过这道题彻底理解了并查集,同时还意识到了我之前一直用map来实现离散化的方法其实是最简单但是最慢的方法,以这道题为例,map导致时间消耗有点大,不吸氧只有70分,吸氧后才能AC。
大体总结了一下,离散化有三种方式
1.直接用map来实现,最简单但是最慢
2.用自己写的HASH函数来实现,最麻烦但是最快
3.去重后排序,并用手写二分查找或lower_bound来获取索引。这个方法不快不慢不简单不复杂,适中。
以后要学习一下后两个方案

Code

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <map>
#include <vector>
#include <queue>
#include <cmath>
using namespace std;

struct DSUs//这不是一个并查集,这是一个元素离散化的不相交集合的集合,即DSUs
{
	private:
	map<int,int> f;
	map<int,int> d;
	inline void insert(int a){if(f.find(a)==f.end())f[a]=a,d[a]=1;}//私有动态初始化函数,不应该被外部访问
	
	public:
	DSUs(){}
	void merge(int a,int b)
	{
		insert(a);insert(b);
		int A=find(a),B=find(b);
		if(A!=B)
		{
			if(d[A]>=d[B])f[B]=A;
			else f[A]=B;
			if(d[A]==d[B])d[A]++;
		}
	}
	inline int find(int a)
	{
		insert(a);
		if(f[a]==a)return a;
		else
		{
			f[a]=find(f[a]);
			return f[a];
		}
	}
	inline void clear()
	{
		f.clear();
		d.clear();
	}
};
DSUs uf;

int n,a[1000000+5],b[1000000+5],c[1000000+5];
int main()
{
	int t;
	std::ios::sync_with_stdio(false);	std::cin.tie(0); 
	cin>>t;
	for(int i=1;i<=t;i++)
	{
		cin>>n;
		for(int j=1;j<=n;j++)
		{
			cin>>a[j]>>b[j]>>c[j];
		}
		for(int j=1;j<=n;j++)
			if(c[j]==1)
				uf.merge(a[j],b[j]);
		int flag=1;
		for(int j=1;j<=n;j++)
		{
			if(c[j]==0)
			{
				if(uf.find(a[j])==uf.find(b[j]))
				{
					flag=0;
					break;
				}
			}
		}
		if(flag==1)cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
		uf.clear();
	}
    return 0;
}