luogu P2184 贪婪大陆 |线段树

题目描述

小FF最后一道防线是一条长度为N的战壕， 小FF拥有无数多种地雷，而SCV每次可以在[ L , R ]区间埋放同一种不同于之前已经埋放的地雷。 由于情况已经十万火急，小FF在某些时候可能会询问你在[ L' , R'] 区间内有多少种不同的地雷， 他希望你能尽快的给予答复。

对于30%的数据： 0<=n, m<=1000;

对于100%的数据： 0<=n, m<=10^5.

输入格式

第一行为两个整数n和m； n表示防线长度， m表示SCV布雷次数及小FF询问的次数总和。

接下来有m行， 每行三个整数Q，L , R； 若Q=1 则表示SCV在[ L , R ]这段区间布上一种地雷， 若Q=2则表示小FF询问当前[ L , R ]区间总共有多少种地雷。

输出格式

对于小FF的每次询问，输出一个答案（单独一行），表示当前区间地雷总数

---

这时候就不难想到用两个线段树维护：

第一个：维护所有节点之前有多少个区间的开头

第二个：维护所有节点之前有多少个区间的结尾

不难证明拿sum[i]-sum[j]得到的就是i~j中间地雷的个数(手动模拟一波就一清二楚了)

#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+10;
inline int read(){
	int x=0; char c=getchar();
	while(c<'0'||c>'9')c=getchar();
	while('0'<=c&&c<='9'){ x=(x<<1)+(x<<3)+(c^48); c=getchar(); }
	return x;
}
int n,m;
#define ls p<<1
#define rs p<<1|1
#define mid ((l(p)+r(p))>>1)
struct node{
struct Seg{
	int l,r,sum,add;
	#define l(x) tree[x].l
	#define r(x) tree[x].r
	#define sum(x) tree[x].sum
	#define add(x) tree[x].add
	#define len(x) (r(x)-l(x)+1)
}tree[N<<2];
void build(int p,int l,int r){
	l(p)=l,r(p)=r;
	if(l==r)return;
	build(ls,l,mid);
	build(rs,mid+1,r);
}
inline void pushdown(int p){
	sum(ls)+=add(p)*len(ls);
	sum(rs)+=add(p)*len(rs);
	add(ls)+=add(p);
	add(rs)+=add(p);
	add(p)=0;
}
void update(int p,int l,int r,int d){
	if(l(p)==r(p)){
		sum(p)+=len(p)*d;
		add(p)+=d;
		return;
	}
	if(add(p))pushdown(p);
	if(l<=mid)update(ls,l,r,d);
	if(r>mid)update(rs,l,r,d);
	sum(p)=sum(ls)+sum(rs);
}
int query(int p,int l,int r){
	if(l<=l(p)&&r(p)<=r)return sum(p);
	if(add(p))pushdown(p);
	int ans=0;
	if(l<=mid)ans+=query(ls,l,r);
	if(r>mid)ans+=query(rs,l,r);
	return ans;	
}
}s,t;
signed main(){
	cin>>n>>m;
	s.build(1,1,n);
	t.build(1,1,n);
	for(int i=1,q,l,r;i<=m;i++){
		q=read(),l=read(),r=read();
		if(q==1){
			s.update(1,l,l,1);
			t.update(1,r,r,1);
		}else{
			if(l>1)printf("%d\n",s.query(1,1,r)-t.query(1,1,l-1));
			else printf("%d\n",s.query(1,1,r));
		}
	}
	
}