线段树模板

线段树是一个区间二叉树,用来处理区间和、区间最大值、区间最小值等区间问题。

在这里插入图片描述

线段树的节点存储着对应的区间信息、以及维护的最大值或者是最小值信息。

线段树是一个完全二叉树,用一个数组来存储,根节点的下标为1,左子节点为2* i,右子节点为2*i+1。

对应的区间信息其实不用存储,都是在代码中维护的,因为每次都是将一个区间按照mid来分为[l,mid]和[mid+1,r]两个左右子区间。

线段树针对的是区间的动态更新和查找,修改和查找的时间复杂度都是log(n)。

因为普通的前缀和的查找时间虽然是O(1),但是更改却是O(n)。

线段树中可以通过惰性更新来增加性能。当更新的时候只更新父节点,只有子节点在被更新或者访问的时候,才会将对应的父节点下推,也就是pushDown,将之前的更新下推到子节点。

下面是一个求区间最大值的线段树模板:

class SegmentTree{
    
    int n;
    int[] tree;
    int[] lazy;
    int[] arr;
    public SegmentTree(int n,int[] arr){
        tree = new int[4 * n];
        lazy = new int[4 * n];
        this.n = n;
        this.arr = arr;
    }
    
    //根据一个arr来初始化线段树
    public void build(int l,int r,int id){
        if(l == r){
            tree[id] = arr[l];
            return;
        }
        
        int mid = (l + r)/2;
        
        build(l,mid,id<<1);
        build(mid+1,r,(id<<1)+1);
        //根据子节点来更新父节点
        pushUp(id);
    }
    
     //根据子节点来更新父节点
    public void pushUp(int id){
        tree[id] = Math.max(tree[id<<1],tree[(id<<1) +1]);
    }
    
     //惰性更新,当要访问父节点的子节点的时候,需要将之前的更新下推给子节点
    public void pushDown(int id){
        tree[id<<1] = Math.max(tree[id<<1],lazy[id]);
        tree[(id<<1)+1] = Math.max(tree[(id<<1)+1],lazy[id]);
        
        lazy[id<<1] = Math.max(lazy[id<<1],lazy[id]);
        lazy[id<<1|1] = Math.max(lazy[(id<<1)+1],lazy[id]);
        
        lazy[id] = 0;
    }
    
    //区间更新
    public void update(int l,int r,int x,int y,int val,int id){
        
        if(x <= l && r <= y){
            tree[id] = Math.max(tree[id],val);
            lazy[id] = Math.max(lazy[id],val);
            return;
        }
        
        int mid = (l + r)/2;
        pushDown(id);
        if(x <= mid){
            update(l,mid,x,y,val,id<<1);
        }
        
        if(y >= mid + 1){
            update(mid+1,r,x,y,val,(id<<1)+1);
        }
        
        pushUp(id);
    }
    
    //区间查询
    public int query(int l,int r,int x,int y,int id){
        if(x <= l && r <= y){
            return tree[id];
        }
        
        int mid = (l + r) / 2;
        int ans = 0;
        pushDown(id);
        if(x <= mid){
            ans = Math.max(ans,query(l,mid,x,y,id<<1));
        }
        
        if(y >= mid + 1){
            ans = Math.max(ans,query(mid+1,r,x,y,(id<<1)+1));
        }
        
        return ans;
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值