题目
思路
修改区间时在开始位置(s+1)表示从则例开始埋一种雷,在结束位置(e+1)表示有一种雷埋到这里就结束了。
查询时输出([1,r])s的和减去([1,l-1])的e的和。用线段树来实现。
(Code)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
const int MAXN=1000001;
int n,m;
struct node{
int l,r,s,e;
}tree[MAXN+1];
inline int read(){
int x=0;bool f=0;char c=getchar();
while(c<'0'||c>'9'){if(c=='-')f=!f;c=getchar();}
while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
return f?-x:x;
}
void build(int l,int r,int now){
tree[now].l=l,tree[now].r=r;
if(l==r){
tree[now].s=0;tree[now].e=0;
return;
}
int mid=(l+r)>>1;
build(l,mid,now<<1);
build(mid+1,r,now<<1|1);
}
void a_s(int x,int now){
if(tree[now].l==tree[now].r){
tree[now].s+=1;
return;
}
int mid=(tree[now].l+tree[now].r)>>1;
if(x<=mid) a_s(x,now<<1);
else a_s(x,now<<1|1);
tree[now].s=tree[now<<1].s+tree[now<<1|1].s;
}
void a_e(int x,int now){
if(tree[now].l==tree[now].r){
tree[now].e+=1;
return;
}
int mid=(tree[now].l+tree[now].r)>>1;
if(x<=mid) a_e(x,now<<1);
else a_e(x,now<<1|1);
tree[now].e=tree[now<<1].e+tree[now<<1|1].e;
}
int ask_s(int x,int y,int now){
if(tree[now].l>=x&&tree[now].r<=y){
return tree[now].s;
}
int mid=(tree[now].l+tree[now].r)>>1,ans=0;
if(x<=mid) ans+=ask_s(x,y,now<<1);
if(y>mid) ans+=ask_s(x,y,now<<1|1);
return ans;
}
int ask_e(int x,int y,int now){
if(tree[now].l>=x&&tree[now].r<=y){
return tree[now].e;
}
int mid=(tree[now].l+tree[now].r)>>1,ans=0;
if(x<=mid) ans+=ask_e(x,y,now<<1);
if(y>mid) ans+=ask_e(x,y,now<<1|1);
return ans;
}
int main(){
n=read(),m=read();
build(1,n,1);
int bz,x,y;
while(m--){
bz=read(),x=read(),y=read();
if(bz==1) a_s(x,1),a_e(y,1);
else printf("%d
",ask_s(1,y,1)-ask_e(1,x-1,1));
}
return 0;
}