线段树板子1(洛谷P3372)

传送

一道线段树板子(最简单的)

似乎之前在培训里写过线段树的样子?不记得了

何为线段树?

一般就是长成这样的树,树上的每个节点代表一个区间。线段树一般用于区间修改,区间查询的问题。

我们如何写一棵线段树?

线段树包含:

1.建树

2.区间修改

3.区间查询与懒标记下传

---------------------------------------------------------------------

一些定义:

sum[k]:节点k所代表的区间的区间和(k是节点编号)

val[i]:在1到n的区间中,点i的权值

laz[k]:在k节点上打的懒标记

1.建树

从根节点开始,递归分别建左子树和右子树,当l=r时,sum[k]=val[l].我们注意到,对于每个不是叶子的节点来说,它的sum值为它的左二字+它的右儿子。同时线段树是一颗二叉树,所以节点k的左儿子的编号就是k*2,右儿子是k*2+1。所以,sum[k]=sum[k*2]+sum[k*2+1]。

 2.区间修改

如果一个区间[l,r]要进行修改,那么与[l,r]有交集的节点都要修改。考虑到与[l,r]有交集的节点数为log(r-l+1)(如果出错欢迎指正),如果直接修改每个点,则复杂度会很高,况且修改了以后还不一定会被查询到。为了降低复杂度,我们采用懒惰的思想。这时候,我们就有了懒标记。

在节点k上打上懒标记,代表k的子树中所有节点都加上laz[k],k节点的sum变为真实值。不过暂时先不真的在左右儿子节点加,如果查询到了,再加。这是待会要讲的标记下传。

所以,对于区间修改来说,我们唯一要做的就是找到被修改区间完全包含的区间,在这个节点上打个懒标记,然后维护一下打上标记的节点的sum(sum[k]+=laz[k]*(r-l+1)),就ok了。

如果当前区间并没有完全被包含,则继续递归寻找。判断它的左右儿子哪个与修改区间有交集,就修改哪个儿子,直到有完全被包含的区间出现。同时,对于当前的这个节点来说,还是要维护sum,sum[k]=sum[k*2]+sum[k*2+1](就是当前节点的sum=左儿子的sum+右儿子的sum)

3.区间查询与懒标记下传

当区间查询的时候,就不能再懒下去了(该干活了),这时候,我们就要把laz[k]扔给它的左儿子,右儿子,让他们变成真实值。

当然,如果没有懒标记(laz[k]=0),那就直接结束了,就不管了。

如果当前区间并不完全被包含在查询区间里面,则递归查询。(要先懒标记下传)看它的哪个儿子与被查询区间有交集,就递归哪个儿子。

也正是因为上述原因,懒标记一次只下传一层(没有传到而要用到的节点会在递归查询中下传到)

在懒标记下传中,把懒标记传给它的左右儿子(不管是否与被查询区间有交集)。让它左右儿子的laz加上laz[k],然后维护他左右儿子的sum。(sum[2*k]+=laz[k]*(mid-l+1),sum[2*k+1]=laz[k]*(r-mid))

这样一棵线段树的基本操作就讲完辣。

代码:

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
const long long N=100009;//小心毒瘤数据范围
long long n,m,val[N*4],sum[N*4],laz[N*4];//注意数组大小
long long read()//读入long long(防毒瘤数据)
{
    char ch=getchar();
    int x=0;bool f=0;
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=(x<<3)+(x<<1)+(ch^48);
        ch=getchar();
    }
    if(f)x=-x;
    return x;
}
int read2()//读入int(为了和函数的参数相匹配)
{
    char ch=getchar();
    int x=0;bool f=0;
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=(x<<3)+(x<<1)+(ch^48);
        ch=getchar();
    }
    if(f)x=-x;
    return x;
}
void zj(int k,int l,int r,int v)//标记下传时候的增加(其实可以写进标记下传函数里面)
{
    laz[k]+=v;
    sum[k]+=v*(r-l+1);
}
void pushdown(int k,int l,int r)//标记下传
{
    if(laz[k]==0)return ;
    long long mid=(l+r)>>1;
     zj(k<<1,l,(int)mid,laz[k]);//位运算优化常数
     zj(k<<1|1,(int)mid+1,r,laz[k]);
    laz[k]=0;
}
void add(int k,int l,int r,int x,int y,int v)//将[x,y]这段区间加上v,l,r为当前递归到的节点代表的区间的左,右端点,k为当前节点编号
{

    if(l>=x&&r<=y)
    {
       laz[k]+=v;
       sum[k]+=v*(r-l+1);
       return;
    }
    long long mid=(l+r)>>1;
    pushdown(k,l,r);
    if(x<=mid)
      add(k<<1,l,(int)mid,x,y,v);
    if(mid<y)
      add(k<<1|1,(int)mid+1,r,x,y,v);
     sum[k]=sum[k<<1]+sum[k<<1|1];// 维护和!!! 
     return;
}
long long query(int k,int l,int r,int x,int y)//查询
{
    if(l>=x&&r<=y)//[l,r]被[x,y]完全覆盖
    {
        return sum[k];
    }
    long long mid=(l+r)>>1,ans=0;
    pushdown(k,l,r);
    if(x<=mid)//判断儿子是否有交集
      ans+=query(k<<1,l,(int)mid,x,y);
    if(mid<y)
       ans+=query(k<<1|1,(int)(mid+1),r,x,y);
    return ans;     
}
void build(int k,int l,int r)//建树
{
    if(l==r)
    {
     sum[k]=val[l];
     return;
    }
    long long mid=(l+r)>>1;
    build(k<<1,l,(int)mid);
    build(k<<1|1,(int)(mid+1),r);
    sum[k]=sum[k<<1]+sum[k<<1|1];
    return;
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;i++)
      val[i]=read();
    build(1,1,n);  
    for(int i=1;i<=m;i++)
    {
        int cz,x,y;
        cz=read2();x=read2();y=read2();
        if(cz==1)//修改
        {
            int k=read2();
            add(1,1,n,x,y,k);
        }
        else//查询
        {
             printf("%lld\n",query(1,1,n,x,y));    
        }
    }  
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/lcez56jsy/p/11105782.html