题意:给出一个长度为n初值为0的数组,以及长度为n的b数组,然后q次操作,add(l,r) 使得区间l~r所有元素+1,或者查询l~r区间a[i]/b[i]的和。
题解:线段树+思维好题,这题构造线段树的方法很巧妙,竟然是以b数组构造,线段树维护的是区间最小值,每次区间加 1变成区间减1,每次更新时,直接将 b 数组的对应区间都减一,如果发现有零,就将对应位置的答案加一并重置回原b数组对应的数,等价于原来的操作,想一想是这么回事。
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
#define ll long long
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));
}
}
return 0;
}