线段树

9 篇文章 0 订阅
1 篇文章 0 订阅
操作(初始化,单个点的修改,区间的修改,区间查询,lazy标记)
//线段树 
//每个节点代表一段区间  除叶节点外均有左右子节点 
//左子节点:[L,(L+R)/2] 右子节点:[(L+R)/2+1,R]  叶节点长度为1 
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int n,a[1001],cnt=1;//cnt树点标号 
struct uio{
    int num,pluslazy,mullazy;//num区间和 pluslazy延迟修改标记(加) mullazy延迟修改标记(乘) 
    int l,r;//该点所表示范围为[l,r] 
    int ls,rs;//左右子节点数组下标 
}tree[2001];//最大不超过2*n 
void pushup(int x)//更新节点信息 
{
    int lson=tree[x].ls;
    int rson=tree[x].rs;
    tree[x].num=tree[lson].num+tree[rson].num;
    tree[x].l=tree[lson].l;
    tree[x].r=tree[rson].r;
}
void pushdown_add(int x)//下载标记(增加)
{
    int lson=tree[x].ls;
    int rson=tree[x].rs;
    tree[lson].num+=tree[x].pluslazy*(tree[lson].r-tree[lson].l+1);
    tree[rson].num+=tree[x].pluslazy*(tree[rson].r-tree[rson].l+1);
    tree[lson].pluslazy+=tree[x].pluslazy;//标记下传 
    tree[rson].pluslazy+=tree[x].pluslazy;//标记下传
    tree[x].pluslazy=0; 
} 
void pushdown_mul(int x)//下载标记(乘积) 
{
    int lson=tree[x].ls;
    int rson=tree[x].rs;
    tree[lson].num*=tree[x].mullazy;
    tree[rson].num*=tree[x].mullazy;
    tree[lson].mullazy*=tree[x].mullazy;//标记下传
    tree[rson].mullazy*=tree[x].mullazy;//标记下传
    tree[lson].pluslazy*=tree[x].mullazy;//标记下传
    tree[rson].pluslazy*=tree[x].mullazy;//标记下传
    tree[x].mullazy=1;
    pushdown_add(x);
} 
void build(int L,int R,int now)//初始化 now当前节点 
{
    if(L==R)//区间长度为1 --> 叶节点 
    {
        tree[now].ls=-1;
        tree[now].rs=-1; 
        tree[now].l=tree[now].r=L;
        tree[now].num=a[L];
        tree[now].mullazy=1;
        return;
    }
    int mid=(L+R)/2;
    tree[now].ls=++cnt;
    tree[now].rs=++cnt; 
    tree[now].mullazy=1;
    build(L,mid,tree[now].ls);
    build(mid+1,R,tree[now].rs);
    pushup(now);//更新节点信息 
} 
void update(int x,int y,int now)//单点修改 x为需修改位置 y为增加值 
{
    if(tree[now].l==tree[now].r)//找到该节点 
    {
        tree[now].num+=y;
        return;
    }
    int mid=(tree[now].l+tree[now].r)/2;
    if(x<=mid)
        update(x,y,tree[now].ls);
    else
        update(x,y,tree[now].rs);
    pushup(now);//更新节点信息 
}
int query(int L,int R,int now)//区间查询 
{
    if(L<=tree[now].l&&tree[now].r<=R)//当前节点代表区间为待查区间的子集 
        return tree[now].num;
    pushdown_mul(now);//下载标记(乘积) 
    int ans=0;
    int mid=(tree[now].l+tree[now].r)/2;
    if(L<=mid)
        ans+=query(L,R,tree[now].ls);
    if(mid+1<=R)
        ans+=query(L,R,tree[now].rs);
    return ans;
}
void update_section_add(int L,int R,int y,int now)//区间修改[L,R](增加)
{
    if(L<=tree[now].l&&tree[now].r<=R)//待查询区间完全覆盖当前节点表示区间 --> 更新区间信息与延迟标记 
    {
        tree[now].num+=y*(tree[now].r-tree[now].l+1);//y*K K为区间长度 
        tree[now].pluslazy+=y;
        return;
    }
    pushdown_mul(now);//下载标记(乘积) 
    int mid=(tree[now].l+tree[now].r)/2;
    if(L<=mid)
        update_section_add(L,R,y,tree[now].ls);
    if(mid+1<=R)
        update_section_add(L,R,y,tree[now].rs);
    pushup(now);//更新节点信息 
}
void update_section_multi(int L,int R,int y,int now)//区间修改[L,R](乘积)
{
    if(L<=tree[now].l&&tree[now].r<=R)//待查询区间完全覆盖当前节点表示区间 --> 更新区间信息与延迟标记 
    {
        tree[now].num*=y;
        tree[now].pluslazy*=y;
        tree[now].mullazy*=y;
        return;
    }
    pushdown_mul(now);//下载标记(乘积) 
    int mid=(tree[now].l+tree[now].r)/2;
    if(L<=mid)
        update_section_multi(L,R,y,tree[now].ls);
    if(mid+1<=R)
        update_section_multi(L,R,y,tree[now].rs);
    pushup(now);//更新节点信息 
} 
void do_something()
{
    return ;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    int root=1; 
    build(1,n,root);//root为根节点 
    do_something();
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值