集合问题(并查集)

本题链接：登录—专业IT笔试面试备考平台_牛客网

题目：

样例1：

4 5 9
2 3 4 5

 样例2：

3 3 4
1 2 4

思路：

        这道题关键点在于。

        当集合中有一个元素均存在于集合 A 和集合 B 的时候是 NO。

        并且  的范围是 1 ~ 1e9 所以，当  >= max(a,b) 的时候也是 NO。

        我们同时可以指定一个 元素范围外的 一个元素作为 根元素集合 A，B

        其次，我们可以将 下标 作为对应的每一个元素，最后进行合并求结果即可。

代码详解如下：

#include <iostream>
#include <vector>
#include <unordered_map>
#define umap unordered_map
#define int long long
#define endl '\n'
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
using namespace std;
 
umap<int,int>pos;	// 存储元素对应的下标
 
// 存储元素集合,至于为什么也用 umap ,由于 Pi 的数据范围上限是 1e9
// 我们要将数组无法开辟这么大，所以我们只能弄个映射 来存储对应的 A，B 根元素
umap<int,int>father;	
 
// 并查集查找函数
inline int Finds(int x)
{
	int t = x;	// 记录其实查找结点
	while(x != father[x]) x = father[x];	// 开始查找
	father[t] = x;	// 路径压缩查找
	return x;	// 返回结果
}
 
// 并查集合并操作
inline void Union(int a,int b)
{
	a = Finds(a),b = Finds(b);	// 查找对应根节点
	father[a] = b;	// 合并对应根节点
}
 
inline void solve()
{
	int n,a,b;
	cin >> n >> a >> b;
	int maxs = max(a,b);	// 获取对应 a b 最大值
	
	int A = maxs + 1;	// 根据对应的最大值，赋值一个元素范围外的元素作为 集合 A 的根节点
	int B = maxs + 2;	// 根据对应的最大值，赋值一个元素范围外的元素并且不同于集合A的根元素的元素作为 集合 B 的根节点
	father[A] = A,father[B] = B;	// 集合根节点初始化
	
	vector<int>v(n + 2,0);	// 存储对应元素
	for(int i = 1;i <= n;++i)
	{
		cin >> v[i];
		if(v[i] >= maxs)	// 如果存在 元素 大于 a 和 b ，那么放不了 任意集合，无解输出 NO
		{
			cout << "NO" << endl;
			return ;
		}
		pos[v[i]] = i;	// 映射对应的下标
		
		father[i] = i;	// 对应下标 根节点初始化
	}
	
	for(int i = 1;i <= n;++i)
	{
		// 如果对应的元素存在的话，我们将其元素的下标与当前的下标进行操作合并对应的集合
		
		if(pos[b - v[i]]) Union(i,pos[b - v[i]]);	// 另一元素存在 集合 b 那么我们合并对应下标 
		else Union(A,i);	//如果不符合那么合并另一个集合
		
		
		if(pos[a - v[i]]) Union(i,pos[a - v[i]]);	// 另一元素存在 集合 a 那么我们合并对应下标 
		else Union(B,i);	//如果不符合那么合并另一个集合
	}
	
	A = Finds(A),B = Finds(B);	// 根据对应的 结合 根节点元素查找；
	
	if(A == B) cout << "NO" << endl;	// 如果最终集合 A 和 集合 B 的根节点也给合并了，说明无解 NO
	else
	{
		cout << "YES" << endl;
		for(int i = 1;i <= n;++i)
		{
// 			cout << bool(Finds(i) == B) << ' ';		// 这样输出是错误的，有可能这里没考虑一个情况，就是 A == B 的时候，也有可能返回值的原因
			if(Finds(i) == A) cout << "0 ";
			else cout << "1 ";
		}
		cout << endl;
	}
}
 
signed main()
{
	IOS;
	int ___t = 1;
	while(___t--) solve();	
	return 0;
}

最后提交：