算法模板——线段树之Lazy标记

一.前言

前面我们已经知道线段树能够进行单点修改和区间查询操作（基本线段树）。那么如果需要修改的是一个区间该怎么办呢？如果是暴力修改到叶子节点，复杂度即为O(nlogn)，显然是十分不优秀的。那么我们能不能向区间查询一样把复杂度降到O(logn)呢？

二.算法流程

线段树肯定是兹瓷O(logn)修改的，否则发明它有何用处？所以，我我们现在需要知道，如何快速进行区间修改操作。首先，我们回顾下终止节点

假定我要在这个图上修改区间[2,8]，我只要修改掉图上所有的终止节点即可。于是复杂度就成功降到O(logn)。
那么，我在终止节点上加什么东西呢？这里，我们就要引进Lazy标记。修改区间的时候，在这个区间所有的终止节点上打上一个标记，代表我这个点要加多少值。
区间查询的时候，如果访问到某个有标记的点，但又不能使用该点的值的时候，就把标记下传（有点绕，看看代码应该就明了了），其他时候直接取值就好

三.例题

1.Pku3468 A Simple Problem with Integers

Description
You have N integers, A1, A2, … , AN. You need to deal with two kinds of operations. One type of op
eration is to add some given number to each number in a given interval. The other is to ask for the
sum of numbers in a given interval.
1．给[a ，b]整体上加上一个常数c。
2．查询[a ，b]区间的和。

Input
The first line contains two numbers N and Q. 1 ≤ N,Q ≤ 100000.
The second line contains N numbers, the initial values of A1, A2, … , AN. -1000000000 ≤ Ai ≤ 1000000000.
Each of the next Q lines represents an operation.
“C a b c” means adding c to each of Aa, Aa+1, … , Ab. -10000 ≤ c ≤ 10000.
“Q a b” means querying the sum of Aa, Aa+1, … , Ab.

Output
You need to answer all Q commands in order. One answer in a line. The sums may exceed the range of 32-bit integers

Sample Input
10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Sample Output
4
55
9
15

HINT
对于100%的数据：N<=100000，M<=100000

区间修改，区间查询，所以用到Lazy标记，代码里会有详细注释

#include<cmath>#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define inf 0x7f7f7f7fusing namespace std;typedef long long ll;typedef unsigned int ui;typedef unsigned long long ull;inline int read(){    int x=0,f=1;char ch=getchar();    for (;ch<'0'||ch>'9';ch=getchar())  if (ch=='-')    f=-1;    for (;ch>='0'&&ch<='9';ch=getchar())    x=(x<<1)+(x<<3)+ch-'0';    return x*f;}inline void print(int x){    if (x>=10)     print(x/10);    putchar(x%10+'0');}const int N=1e5;int n,m;ll val[N+10];struct Segment{    #define ls (p<<1)    #define rs (p<<1|1)    ll tree[N*4+10],cnt[N*4+10];    void updata(int p){tree[p]=(tree[ls]+tree[rs]);}    //updata的内容具体情况具体分析     void add_cnt(int p,int v,int t){    //标记不影响当前节点的值，谨记这点         tree[p]=tree[p]+1ll*v*t;        //t是区间长度，加标记需要乘上这个区间         cnt[p]=cnt[p]+v;                //标记更新     }    void pushdown(int p,int l,int r){   //加标记下传         if (!cnt[p])    return;        int mid=(l+r)>>1;        add_cnt(ls,cnt[p],mid-l+1),add_cnt(rs,cnt[p],r-mid);    //加标记的更新         cnt[p]=0;    }    void build(int p,int l,int r){        cnt[p]=0;        if (l==r){            tree[p]=val[l];     //val[i]基本是开始读入的数            return;        }        int mid=(l+r)>>1;        build(ls,l,mid),build(rs,mid+1,r);        updata(p);    }    void change(int p,int l,int r,int x,int y,int t){   //t是标记         if (x<=l&&r<=y){            add_cnt(p,t,r-l+1); //打加标记             return;        }        int mid=(l+r)>>1;        pushdown(p,l,r);    //修改两边的时候记得传递标记         if (x<=mid) change(ls,l,mid,x,y,t);         //访问左儿子         if (y>mid)  change(rs,mid+1,r,x,y,t);       //访问右儿子         updata(p);      //记得更新     }    ll query(int p,int l,int r,int x,int y){        if (x<=l&&r<=y) return tree[p];     //在区间内直接输出         int mid=(l+r)>>1;        ll ans=0;        pushdown(p,l,r);    //传递标记         if (x<=mid) ans=ans+query(ls,l,mid,x,y);    //访问左儿子         if (y>mid)  ans=ans+query(rs,mid+1,r,x,y);  //访问右儿子         return ans;    }}Tree;char s;int main(){    n=read(),m=read();    for (int i=1;i<=n;i++)  val[i]=read();    Tree.build(1,1,n);    for (int i=1;i<=m;i++){        cin>>s;        int x=read(),y=read(),z;        if (s=='Q') printf("%lld\n",Tree.query(1,1,n,x,y));        if (s=='C') z=read(),Tree.change(1,1,n,x,y,z);    }    return 0;}

2.

Description
给定一个正整数序列A，要求支持以下操作
1）： + a b c 表示在[a，b]上加上一个常数C。
2）： * a b c 在[a，b]上乘上一个常数K。
3）： QUERY a b 查询[a，b]的sum。

Input
第一行两个正整数n、m，n表示序列长度，m表示操作数
第二行n个正整数，第i表示A[i]的大小
接下来的m行，每行有且仅有一种操作，具体和题目描述一致?
n,m<=100000
其他权值都<=50000

Output
对于每个询问操作，输出答案对1000000007取余的结果

Sample Input
10 10
50 14 20 18 19 11 43 43 26 44
+ 3 6 41
QUERY 1 2
+ 5 5 14
QUERY 1 3
QUERY 4 4
QUERY 2 6
* 3 5 31
* 4 8 20
* 5 8 28
QUERY 6 7

Sample Output
64
125
59
260
53200

同样是区间修改，区间查询，不过本题比上题要难一些。为什么呢，因为它需要维护两个标记。
如果是两个互不相干的标记还好说，但这两个标记互相干扰！！！
因此，我们在处理这类有多个标记的题目的时候，需要考虑下这些标记互相之间的影响，标记与标记之间的先后更新顺序。
如本题，加标记与乘标记势必会互相影响
如果我们先更新加标记，那么乘标记就会变成一个实型，非常不方便。于是我们让乘标记先更新，加标记就只需要乘上一个乘标记即可，大大降低了实现难度。
（注释基本没有，不过也请读者凑合着看吧）

#include<cmath>#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define inf 0x7f7f7f7fusing namespace std;typedef long long ll;typedef unsigned int ui;typedef unsigned long long ull;inline int read(){    int x=0,f=1;char ch=getchar();    for (;ch<'0'||ch>'9';ch=getchar())    if (ch=='-')    f=-1;    for (;ch>='0'&&ch<='9';ch=getchar())  x=(x<<1)+(x<<3)+ch-'0';    return x*f;}inline void print(int x){    if (x>=10)     print(x/10);    putchar(x%10+'0');}const int N=1e5,mod=1e9+7;int n,m;struct Segment{    #define ls (p<<1)    #define rs (p<<1|1)    int tree[N*4+10],cnt[N*4+10],res[N*4+10],size[N*4+10];    void updata(int p){tree[p]=(tree[ls]+tree[rs])%mod;}    void add_cnt(int p,int v){        tree[p]=(tree[p]+1ll*v*size[p])%mod;        cnt[p]=(cnt[p]+v)%mod;    }    void pushdown_cnt(int p){//加标记下传        if (!cnt[p])    return;        add_cnt(ls,cnt[p]),add_cnt(rs,cnt[p]);        cnt[p]=0;    }    void add_res(int p,int v){        tree[p]=1ll*tree[p]*v%mod;        res[p]=1ll*res[p]*v%mod;        cnt[p]=1ll*cnt[p]*v%mod;    }    void pushdown_res(int p){//乘标记下传        if (res[p]==1)  return;        add_res(ls,res[p]),add_res(rs,res[p]);        res[p]=1;    }    void pushdown(int p){pushdown_res(p),pushdown_cnt(p);}    void build(int p,int l,int r){        cnt[p]=0,res[p]=1,size[p]=r-l+1;        if (l==r){            tree[p]=read();            return;        }        int mid=(l+r)>>1;        build(ls,l,mid),build(rs,mid+1,r);        updata(p);    }    void change(int p,int l,int r,int x,int y,int Cnt,int Res){        if (x<=l&&r<=y){            add_res(p,Res),add_cnt(p,Cnt);            return;        }        int mid=(l+r)>>1;        pushdown(p);        if (x<=mid)  change(ls,l,mid,x,y,Cnt,Res);        if (y>mid)   change(rs,mid+1,r,x,y,Cnt,Res);         updata(p);    }    int query(int p,int l,int r,int x,int y){        if (x<=l&&r<=y)   return tree[p];        int mid=(l+r)>>1,ans=0;        pushdown(p);        if (x<=mid)  ans=(ans+query(ls,l,mid,x,y))%mod;        if (y>mid)   ans=(ans+query(rs,mid+1,r,x,y))%mod;        return ans;    }}Tree;char s[10];int main(){    n=read(),m=read();    Tree.build(1,1,n);    for (int i=1;i<=m;i++){        scanf("%s",s+1);        int x=read(),y=read(),z;        if (s[1]=='Q')  printf("%d\n",Tree.query(1,1,n,x,y));        if (s[1]=='+')  z=read(),Tree.change(1,1,n,x,y,z,1);        if (s[1]=='*')  z=read(),Tree.change(1,1,n,x,y,0,z);    }    return 0;}

3.最大连续子数列和