线段树整理

笑着哭i 提交于 2020-01-18 09:05:37

仅用于个人复习 有问题还望指出

线段树 是一个基于二分思想的一种二叉搜索树 能在 log2(n) 的时间内 完成更新和查找操作 其中 线段树有以下操作
1.建树
线段树的模板大都是固定的 最为重要的是 应该明确用线段树来维护什么信息 也就是你得到答案需要的信息
明确了这个信息之后 之后的操作就会显得行云流水

struct node///结构体保存信息
{
    int l,r,ans;
}t[maxn*4];
void build(int v,int l,int r)
{
    t[v].l=l;
    t[v].r=r;
    if(l==r)
    {
        scanf("%d",&t[v].ans);
        return ;
    }
    int mid=(l+r)/2;
    build(2*v,l,mid);//建立左子树
    build(2*v+1,mid+1,r);//右子树
    t[v].ans=max(t[v*2].ans,t[v*2+1].ans);///更新最大值
}

左子树和右子树 也可以使用位运算的版本 分别为 v<<1 和 v<<1|1
线段树的内存一般要开大4倍左右 N个叶节点的满二叉树 有2N-1个结点 还有最后一层有空余 所以需要4N的空间来保证不会越界

2.单点更新

void update(int v,int x,int y)///把结点x赋值成y
{
    if(t[v].l==t[v].r)
    {
        t[v].ans=y;
        return ;
    }
    int mid=(t[v].l+t[v].r)/2;
    if(x<=mid)
        update(v*2,x,y);
     else
        update(v*2+1,x,y);
     t[v].ans=max(t[v*2].ans,t[v*2+1].ans);///回溯状态时也需要更新
}

其中if条件内使用 t[v].l == x && t[v].r == x也是一样的

3.单点查找
单点查找也就跟单点更新差不多 但是不需要更新结点上的值 访问就行

4.区间更新

区间更新 如果使用每个结点都更新过去复杂度会来到O(n) 所以采用了一种叫lazy延迟标记的办法 当这个区间需要更新时 先将其加上lazy标记 如果后面遇到需要更新或者查询时再次访问到这个结点时 再将区间进行更新

关键点 在于 要明确lazy对于区间的影响 和 更新

void push(int v)
{
    t[v<<1].sum=(t[v<<1].r-t[v<<1].l+1)*t[v].lazy;
    t[v<<1|1].sum=(t[v<<1|1].r-t[v<<1|1].l+1)*t[v].lazy;
    t[v<<1].lazy=t[v].lazy;
    t[v<<1|1].lazy=t[v].lazy;
    t[v].lazy = 0;
}

void update(int v,int x,int y,int value)
{
    if(t[v].l==x&&t[v].r==y)
    {
        t[v].sum=(y-x+1)*value;
        t[v].lazy=value;
        return ;
    }
    if(t[v].lazy)
    {
        push(v);
    }
    int mid=(t[v].l+t[v].r)/2;
    if(y<=mid)
        update(v*2,x,y,value);
    else
    {
        if(x>mid)
            update(v*2+1,x,y,value);
        else
        {
            update(v*2,x,mid,value);
            update(v*2+1,mid+1,y,value);
        }
    }
    t[v].sum=t[v*2].sum+t[v*2+1].sum;
}

在update函数中 还有一种对于区间的操作时 这个区间可能是横跨左右子树的 所以对左右子树都需要 更新操作 对于询问操作等 也是一样的

5.区间查询

void query(int v,int l, int r)
{
    if(a[v].l>r||a[v].r<l)
        return ;
    if(a[v].l>=l&&a[v].r<=r)
    {
        ans+=a[v].sum;
        return ;
    }
       if(a[v].add)
    {
        a[v*2].sum+=(a[v*2].r-a[v*2].l+1)*a[v].add;
        a[v*2].add+=a[v].add;
        a[v*2+1].sum+=(a[v*2+1].r-a[v*2+1].l+1)*a[v].add;
        a[v*2+1].add+=a[v].add;
        a[v].add=0;
    }
    query(2*v,l,r);
    query(2*v+1,l,r);
    a[v].sum=a[v*2].sum+a[v*2+1].sum;
}

区间查询也需要注意跨区间的操作

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!