线段树懒标记详解
懒标记顾名思意,就是一个很懒的标记,怎么个懒法呢,就是你不需要展开那个区间,它就永远不会起到作用,这也就是线段树在区间查询可以做到 logn 时间复杂度的原因,因为它对原本 区间上所有数值的改变,变为了对区间添加一个标记,后续可能会展开,也可能不再展开,这也就是线段树的懒标记节省时间的关键
我们这里要展开的线段树懒标记的写法,是一种懒标记在这一区间但这一区间已经做出了相应的改变完后的状态,也就是说懒标记在的这一层是用来存的它的左右孩子需要做出的变化,自己已经变完了
而且我们要做到懒标记的统一,也就是一个区间上的add值是一样的,这也就要求我们在改变区间值时,如果要分开一个区间处理,一定要先pushdown后再修改,以保证这个区间add值得统一,不能一段是一个数,另一段时另一个数,而且我们在查询时,也就是开头提到的,我们需要分开区间展开时就需要把懒标记pushdown下去来获取我们需要得区间和。
链接:AcWing_243
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
typedef long long ll;
struct node{
int l,r;
ll sum,add;
}tr[N<<2];
ll arr[N];
int n,m;
void pushup(int u)
{
node &a=tr[u];
node &l=tr[u<<1];
node &r=tr[u<<1|1];
a.sum=l.sum+r.sum;
}
void pushdown(int t)
{
node &u=tr[t];
node &l=tr[t<<1];
node &r=tr[t<<1|1];
if(u.add)
{
l.add+=u.add;
l.sum+=(l.r-l.l+1)*u.add;
r.add+=u.add;
r.sum+=(r.r-r.l+1)*u.add;
u.add=0;
}
}
void build(int u,int l,int r)
{
if(l==r)
{
tr[u]={l,r,arr[l],0};
}
else
{
tr[u].l=l,tr[u].r=r;
int mid=l+r>>1;
build(u<<1,l,mid);
build(u<<1|1,mid+1,r);
pushup(u);
}
}
void modify(int u,int l,int r,int x)
{
if(tr[u].l>=l&&tr[u].r<=r)
{
tr[u].sum+=(tr[u].r-tr[u].l+1)*x;
tr[u].add+=x;
}
else
{
pushdown(u);
int mid=tr[u].l+tr[u].r>>1;
if(l<=mid) modify(u<<1,l,r,x);
if(r>mid) modify(u<<1|1,l,r,x);
pushup(u);
}
}
ll query(int u,int l,int r)
{
//第一个进去的区间是u=1,即整个区间,所以会先pushdown然后再找符合条件的区间返回
if(tr[u].l>=l&&tr[u].r<=r)
{
return tr[u].sum;
}
pushdown(u);
ll res=0;
int mid=tr[u].l+tr[u].r>>1;
if(l<=mid) res+=query(u<<1,l,r);
if(r>mid) res+=query(u<<1|1,l,r);
return res;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
scanf("%lld",&arr[i]);
}
build(1,1,n);
while(m--)
{
char op[2];
int l,r,x;
scanf("%s%d%d",op,&l,&r);
if(*op=='Q')
{
printf("%lld\n",query(1,l,r));
}
else
{
scanf("%d",&x);
modify(1,l,r,x);
}
}
return 0;
}