湖南省第十四届程序设计大赛 H.千万别用树套树(思维题+容斥)

题目

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

Bobo 精通数据结构！他想维护一个线段的集合 S。初始时，S 为空。他会依次进行 q 次操作，操作有 2 种。
* 类型 1：给出 l, r，向集合 S 中插入线段 [l, r].
* 类型 2：给出 l, r，询问满足 [x,y]∈S[x, y] 且 x≤l≤r≤y 的线段 [x, y] 数量。
帮 Bobo 求出每次询问的答案。

数据组数不超过10，1<=n,q<=1e5，

保证对于类型2的询问，[l,r]区间长度不超过3，即r-l<=2

思路来源

登录—专业IT笔试面试备考平台_牛客网

题解

容斥，为了得到左端点<=l，右端点>=r的线段，

首先，用左端点<=l的线段数，减去右端点>=r-1的线段

会得到一个值，这个值代表的含义是：

左端点<=l且右端点>=r的线段数减去左端点>=l+1且右端点>=r的线段数

说明多减的部分是左端点>=l+1，而右端点<=r-1的线段数，

再把这部分加回来即可

注意到r-l<=2，这种情况只会在r-l=2且l+1==r-1的时候会出现，

所以，单独开一个数组c，专门用来记录长度为1的线段

可以用以下这个例子，结合理解：

4 2
1 4 4
2 3 5

代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=1e5+10;
int n,q,op,l,r;
int a[N],b[N],c[N],ans;
void add(int *tr,int x,int v)
{
	for(int i=x;i<=n;i+=i&-i)
	tr[i]+=v;
}
int sum(int *tr,int x)
{
	int ans=0;
	for(int i=x;i>0;i-=i&-i)
	ans+=tr[i];
	return ans;
}
int main()
{
	while(~scanf("%d%d",&n,&q))
	{
        for(int i=1;i<=n;++i)
            a[i]=b[i]=c[i]=0;
		for(int i=1;i<=q;++i)
		{
			scanf("%d%d%d",&op,&l,&r);
			if(op==1)
            {
                add(a,l,1),add(b,r,1);
                if(l==r)c[l]++;
            }
            else
            {
                int ans=sum(a,l)-sum(b,r-1);
                if(r-l==2)ans+=c[l+1];
                printf("%d\n",ans);
            }
        }
	}
	return 0;
}