线段树下线段果,线段树下你和我,线段树下玩游戏,懒惰更新花样多。多校二第七题,昨天竟然没想出来,这题构造线段树的方法十分巧妙,竟然是以b数组构造,线段树维护的是区间最小值,每次区间加 1 变成区间减 1 ,然后当区间最小值即将成为0的时候,就要一直更新到最小值为0的叶子节点,然后0的个数就加1,再把为0的那个节点再改回成 b 数组对应的值,然后区间查询就查询区间有多少个0即可。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
typedef long long LL;
struct node
{
int _min,val,sum,add;
}a[maxn*3];
void pushdown(int o,int ls,int rs)
{
a[ls].add+=a[o].add;
a[rs].add+=a[o].add;
a[ls]._min+=a[o].add;
a[rs]._min+=a[o].add;
a[o].add=0;
}
void pushup(int o,int ls,int rs)
{
a[o]._min=min(a[ls]._min,a[rs]._min)+a[o].add;
a[o].sum=a[ls].sum+a[rs].sum;
}
void build(int l,int r,int o)
{
a[o].sum=a[o].add=0;
if(l==r)
{
scanf("%d",&a[o].val);
a[o]._min=a[o].val;
return;
}
int ls=o*2,rs=o*2+1,m=(l+r)/2;
build(l,m,ls);
build(m+1,r,rs);
pushup(o,ls,rs);
}
void update(int l,int r,int o,int ql,int qr)
{
if(a[o]._min>1&&l>=ql&&r<=qr)
{
a[o].add--;
a[o]._min--;
return;
}
if(a[o]._min==1&&l==r)
{
a[o].sum++;
a[o]._min=a[o].val;
a[o].add=0;
return;
}
int ls=o*2,rs=o*2+1,m=(l+r)/2;
if(a[o].add!=0)
pushdown(o,ls,rs);
if(ql<=m)
update(l,m,ls,ql,qr);
if(qr>m)
update(m+1,r,rs,ql,qr);
pushup(o,ls,rs);
}
LL query(int l,int r,int o,int ql,int qr)
{
if(l>=ql&&r<=qr)
return a[o].sum;
int ls=o*2,rs=o*2+1,m=(l+r)/2;
LL res=0;
if(ql<=m)
res+=query(l,m,ls,ql,qr);
if(qr>m)
res+=query(m+1,r,rs,ql,qr);
return res;
}
int main()
{
int n,m,ql,qr;
char s[100];
while(~scanf("%d%d",&n,&m))
{
build(1,n,1);
while(m--)
{
scanf("%s%d%d",s,&ql,&qr);
if(s[0]=='a')
update(1,n,1,ql,qr);
else
printf("%lld\n",query(1,n,1,ql,qr));
}
}
}