Restructuring Company（区间合并并查集）

题目地址

题意

对于一个序列1，2，3…n。有三个操作
1.合并x，y
2.合并x，x+1，x+2…y
3.查询x和y是否在一个集合

思路

并查集维护
操作1：直接合并即可
操作2：维护一个a[i],表示元素i所在集合的右边界
这个右边界有两种情况

例如1，3，4，5在一个集合中，那a[1]=1，如果1，2，3，4在一个集合中，那a[1]=4

代码

#include<bits/stdc++.h>
#include<array>
#include<unordered_map>
using namespace std;
typedef long long ll;
const int N = 1e6 + 5;
ll n, m;
ll k;
ll a[N];
int t = 1;
ll f[N];
int find(int x) {
	if (x == f[x])return f[x];
	return f[x] = find(f[x]);
}
void solve()
{
	cin >> n >> m;
	for (int i = 1; i <= n; i++)a[i]=f[i] = i;
	for (int k = 1; k <= m; k++) {
		int id;
		cin >> id;
		ll x, y;
		cin >> x >> y;
		if (id == 1) {
			f[find(x)] = find(y);
		}
		else if (id == 2) {
			a[x] = a[y];
			for (int i = x; i < y;) {
				f[(find(i+1))] = find(i);
				//区间右端点直接变成a[y]
				int tem = a[i + 1];
				a[1+i] = a[y];
				i = tem;
			}
		}
		else {
			cout<<( find(x) == find(y) ? "YES" : "NO")<<endl;
		}
	}
	
}
int main()
{
	
	std::ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);

	//cin >> t;
	while (t--) {
		solve();
	}
}