题目描述
小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
",s.query(1,1,r)-t.query(1,1,l-1));
else printf("%d
",s.query(1,1,r));
}
}
}