这里简单的讲下懒惰标记,懒惰标记即是要修改一个区间的时候,暂时不需要修改区间中的每个点,加一个标志代表这个区间被修改了,等到需要查询一个小区间的时候再具体修改。至于为什么要这样呢,不妨这么考虑,当更新一段区间的时候,如果没有懒惰标记,一开始如果找到的区间比要更新的区间大,那么这个大的区间里的每个点都需要修改,这样时间复杂度就是o(n)了,这跟没用线段树一样,至于单点更新为什么不需要这个呢,是因为单点更新都是更新一个点,所以不会出现更新区间这样的问题,而采用了懒惰标记,只是对当前父亲节点进行标记,当找到更新的区间时,才会进行更新,而不是在每次找区间的时候都更新,这样的时间复杂度才是线段树的复杂度。
View Code
#include<iostream>
#include<cstdio>
using namespace std;
#define maxn 100100
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
long long sum[maxn<<2],flag[maxn<<2];
void Pushup(int rt) //向上专递sum值
{
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void Pushdown(int rt,int num) //向下修改,更新左右孩子
{
flag[rt<<1]+=flag[rt];
flag[rt<<1|1]+=flag[rt];
sum[rt<<1]+=(num-(num>>1))*flag[rt]; //左孩子的区间长度大于等于右孩子
sum[rt<<1|1]+=(num>>1)*flag[rt];
flag[rt]=0; //修改后,把延迟标记置0
}
void build(int l,int r,int rt) //建树
{
flag[rt]=0;
if(l==r)
{
scanf("%lld",&sum[rt]);
return ;
}
int m=(l+r)>>1;
build(lson); //建左孩子
build(rson); //建右孩子
Pushup(rt); //向上更新sum值
}
void update(int L,int R,int w,int l,int r,int rt) //修改区间的值
{
if(L<=l&&r<=R)
{
flag[rt]+=w; //修改的增量是w
sum[rt]+=(long long)(r-l+1)*w; //区间长度为(r-l+1),每个元素都要加w
return;
}
Pushdown(rt,r-l+1); //向下修改
int m=(l+r)>>1;
if(L<=m) update(L,R,w,lson);
if(R>m) update(L,R,w,rson);
Pushup(rt); //修改完,向上传递sum
}
long long query(int L,int R,int l,int r,int rt) //查询区间的sum
{
if(L<=l&&r<=R)
return sum[rt];
Pushdown(rt,r-l+1); //向下边修改边查询
long long ret=0;
int m=(l+r)>>1;
if(L<=m) ret+=query(L,R,lson);
if(R>m) ret+=query(L,R,rson);
return ret;
}
int main()
{
int n,m,a,b,w;
char c[2];
while(~scanf("%d %d",&n,&m))
{
build(1,n,1);
while(m--)
{
scanf("%s",&c);
if(c[0]=='Q')
{
scanf("%d %d",&a,&b);
printf("%lld\n",query(a,b,1,n,1));
}
else
{
scanf("%d %d %d",&a,&b,&w);
update(a,b,w,1,n,1);
}
}
}
return 0;
}
(r - l + 1) >> 1是右儿子的区间长度。
实际上(r - l + 2) >> 1是(r - l + 1) / 2.0的向上取整。注意由于在build中,mid归于左儿子,
故左儿子的区间长度>=右儿子区间长度;