CodeCraft-22 and Codeforces Round #795 (Div. 2) E. Number of Groups

题目链接

题意

给定一个数轴，数轴上有 $n$ 条线段，线段有两种颜色，如果两个不同颜色的线段有交集，那么他们就属于一个集合 ，问最后有多少个集合。

题解

将红色的线段以左端点为关键字升序，蓝色的以右端点升序，那么两种不同颜色线段有交集就是红色左端点小于等于蓝色右端点的情况下，红色的右端点大于等于蓝色的左端点。

因此我们需要对每个蓝色的线段，对满足左端点小于等于该蓝色右端点的红色线段分情况讨论

1. 如果红色线段的右端点大于等于该蓝色的左端点的话，那么它两就合并到一起，如下图的 $2、5$
2. 否则就把红色线段放到一个以右端点降序单调队列里，因为这个红色线段可能会和后面的蓝色线段有交集，如下图的 $1$ 不能和 $5$ 产生交集，但是 $1、6$ 可以合并

处理完新的红色线段后，还需要和单调队列里的线段合并。单调队列中的左端点必定小于蓝色线段的右端点，因此只需一直判断到队列中红色线段的右端点小于该蓝色的左端点后就可以退出了。

还有一个关键，前面被前面的蓝色线段合并的红色线段也可能和当前的蓝色合并，如 $2、5$ 合并，同时 $2、6$ 也能合并，但是 $2$ 不在单调队列中。此时我们就得考虑前面的蓝色线段是否能间接的和当前蓝色线段合并。

我们可以设一个 $mx$，表示能和当前蓝色线段合并的红色线段的最大右端点，同时加入到单调队列中，这样我们在后面的蓝色线段处理中，可以在处理单调队列时与其合并。如对 $5$ 来说，$mx=2$，那么就把 $(5,2)$ 加入单调队列，当处理 $6$ 时，可以在单调队列中取出 $(5,2)$ 满足条件，因此合并 $5、6$

代码

#include<bits/stdc++.h>
#define Ppi pair<pair<int,int>,int>
#define Pii pair<int,int>
#define l first.first
#define r first.second
#define fi first
#define se second
#define mp make_pair
#define pb push_back
using namespace std;
const int N=1e5+9;
int T,n,ans;
int f[N],vis[N];
vector<Ppi>a[2];
priority_queue<Pii>q;
int find(int x)
{
	return f[x]==x?x:f[x]=find(f[x]);
}
void solve()
{
	cin>>n;
	a[0].clear(); a[1].clear();
	for(int i=1,op,x,y;i<=n;i++)
	{
		cin>>op>>x>>y;
		a[op].pb(mp(mp(x,y),i));
		f[i]=i; vis[i]=0;
	}
	sort(a[0].begin(),a[0].end(),[](Ppi x,Ppi y){return x.fi.se<y.fi.se;});
	sort(a[1].begin(),a[1].end(),[](Ppi x,Ppi y){return x.fi.fi<y.fi.fi;});
	int i=0,sz=a[1].size();
	while(!q.empty()) q.pop();
	for(Ppi x:a[0])
	{
		int mx=0;
		while(i<sz&&x.r>=a[1][i].l)
		{
			if(a[1][i].r>=x.l)
			{
				f[find(x.se)]=find(a[1][i].se);
				mx=max(mx,a[1][i].r);
			}
			else q.push(mp(a[1][i].r,a[1][i].se));
			i++;
		}
		while(!q.empty())
		{
			Pii y=q.top();
			if(y.fi<x.l) break;
			f[find(x.se)]=find(y.se);
			mx=max(mx,y.fi);
			q.pop();
		}
		if(mx) q.push(mp(mx,x.se));
	}
	ans=0;
	for(int i=1;i<=n;i++)
	{
		int fa=find(i);
		if(vis[fa]) continue;
		vis[fa]=1;
		ans++;
	}
	cout<<ans<<"\n";
}
int main()
{
	ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
	cin>>T;
	while(T--) solve();
	return 0;
}