当前位置:网站首页>动态线段树leetcode.715

动态线段树leetcode.715

2022-06-26 03:16:00 路Lu727

private Map<Integer, Integer> tree;//线段树,表示某节点的值
    private Map<Integer, Integer> lazy;//懒标记
    private int N;
    public RangeModule() {
        tree = new HashMap<Integer, Integer>();
        lazy = new HashMap<Integer, Integer>();
        N= (int)1e9;
    }
    //区间修改(增加)
    public void update(int l,int r,int val){
        if(l>r) return;
        update(l,r,1,N,1,val);
    }
    private void update( int l, int r,int curl, int curr, int node,int val) {
        //区间无交集
        if (curl>r||curr<l)
            return;
        //当前区间在目标区间内
        if (l<=curl&&curr<=r) {
            int sum=tree.getOrDefault(node, 0);//当前区间和
            if(val==1&&sum<(curr-curl+1)*val){//当前区间部分节点未跟踪
                tree.put(node,(curr-curl+1)*val);//更新当前节点
                if(curl<curr) lazy.put(node, 1);
            }else if(val==-1&&sum>0) {//当前区间存在已跟踪节点
                tree.put(node, 0);//更新当前节点
                if (curl < curr) lazy.put(node, -1);
            }
        } else {//当前区间与目标区间存在交集
            int mid = (curl + curr) >> 1;
            if(lazy.getOrDefault(node,0)!=0)
                pushDown(node,curr-curl+1);
            update( l, r, curl, mid,2 * node,val);
            update(l, r, mid+1, curr, 2 * node + 1,val);
            tree.put(node, tree.getOrDefault(2 * node, 0)+tree.getOrDefault(2 * node + 1, 0));
        }
    }
    //将标记向下一层传递
    private void pushDown(int node,int len){
        int _lazy=lazy.get(node);
        int sum1=tree.getOrDefault(node*2,0);
        int sum2=tree.getOrDefault(node*2+1,0);
        if(_lazy==1){
            //区间存在未跟踪节点
            if(sum1<(len-len/2)*_lazy){
                lazy.put(node*2,1);
                tree.put(node*2,(len-len/2)*_lazy);
            }
            if(sum2<len/2*_lazy){
                lazy.put(node*2+1,1);
                tree.put(node*2+1,len/2*_lazy);
            }
        }else{
            //区间存在已跟踪节点
            if(sum1>0){
                lazy.put(node*2,-1);
                tree.put(node*2,0);
            }
            if(sum2>0){
                lazy.put(node*2+1,-1);
                tree.put(node*2+1,0);
            }
        }
        lazy.put(node,0);
    }
    //区间查询
    public int query(int l,int r){
        if(l>r) return -1;
        return query(l,r,1,N,1);
    }
    private int query(int l,int r,int curl,int curr,int node){
        //不在目标区间
        if(curl>r||curr<l)
            return 0;
        //在目标区间
        if(l<=curl&&curr<=r)
            return tree.getOrDefault(node,0);
        //部分有交集,分割区间处理
        int mid=(curl+curr)>>1;
        if(lazy.getOrDefault(node,0)!=0)
            pushDown(node,curr-curl+1);
        return  query(l,r,curl,mid,node*2)+query(l,r,mid+1,curr,node*2+1);
    }
 
    public void addRange(int left, int right) {
        update( left, right - 1, 1);
    }
    public boolean queryRange(int left, int right) {
        return query(left, right - 1) == right - left;
    }
    public void removeRange(int left, int right) {
        update(left, right - 1, -1);
    }

原网站

版权声明
本文为[路Lu727]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_60466670/article/details/125452697