P2161 [SHOI2009]会场预约（区间set）

题目描述

你需要维护一个 在数轴上的线段 的集合 S，支持两种操作：
A l r 表示将 S 中所有与线段 [l,r] 相交的线段删去，并将 [l,r] 加入 S 中。
B 查询 S 中的元素数量。
对于 A 操作，每次还需输出删掉的元素个数。

输入格式

第一行一个正整数 n，表示操作个数。
接下来 n 行，每行表示一个操作，都是上面两种中的一个。

输出格式

输出 n 行，每行一个整数，表示对应操作的答案。

输入输出样例

输入
6
A 10 15
A 17 19
A 12 17
A 90 99
A 11 12
B
输出
0
0
2
0
1
2

【数据范围】

对于 100% 的数据，1 ≤ n ≤ 2×10⁵，1 ≤ l ≤ r ≤ 10⁵。

题目分析

这个题需要我们维护一个区间不相交的集合。因为set集合具有自动去重复的特点，因此我们可以考虑用set集合来维护该集合。

那么如何定义<号才能够使得相交的区间相等呢?

bool operator<(const Node &a)const
{ return r<a.l; }

这样定义<号，使得a<b表示a区间完全在b的左边，a>b表示a区间完全在b的右边，a=b表示a区间和b相交。

然后用set来维护这个集合就可以了。

代码如下

#include <iostream>
#include <cmath>
#include <cstdio>
#include <set>
#include <unordered_set>
#include <string>
#include <cstring>
#include <map>
#include <unordered_map>
#include <algorithm>
#include <stack>
#include <queue>
#include <bitset>
#define LL long long
#define PII pair<LL,int>
#define PDD pair<double,double>
#define x first
#define y second
using namespace std;
const int N=1e5+5,INF=0x3f3f3f3f;
struct Node {					//定义区间结构体
	int l,r;
	bool operator<(const Node &a)const
	{ return r<a.l; }
};
set<Node> s;
int main()
{
	int n;
	scanf("%d",&n);
	while(n--)
	{
		char op[2];
		scanf("%s",op);
		if(op[0]=='A')
		{
			int l,r;
			scanf("%d%d",&l,&r);
			int cnt=0;
			auto k=s.find({l,r});	//在将[l,r]区间加入set之前，先将与[l,r]相等的区间全部从集合中删除
			while(k!=s.end())
			{
				cnt++;				//同时记录删除的个数
				s.erase(k);
				k=s.find({l,r});
			}
			s.insert({l,r});
			printf("%d\n",cnt);
		}
		else printf("%d\n",s.size());	//输出集合中元素个数即可
	}
	return 0;
}