Segment Tree 线段树算法(java)

线段树算法

  • Segment Tree 线段树算法
    • 代码演示
  • 蓄水池算法

Segment Tree 线段树算法

什么是线段树算法:
线段树(Segment Tree)是一种基于树结构的数据结构,用于解决区间查询问题,例如区间最大值、最小值、区间和等。线段树是一种高度平衡的二叉树,每个节点都代表了一个区间。下面我们来详细了解一下线段树算法。

线段树的构建
线段树的构建过程可以通过递归的方式实现。对于一个给定的数组,我们首先构建一个高度为 log n 的线段树,其中 n 是数组的长度。每个节点都代表了一个区间,例如左儿子节点代表[l, r],右儿子节点代表[r+1, r2]。具体的构建过程如下:
1.初始化根节点,区间为[0, n-1]。
2.对于每个节点,如果它的左右儿子节点存在,就递归构建左右儿子节点。
3.每个节点的值根据需要更新,例如查询最大值时,每个节点存储的值应该是该区间内的最大值
4.当构建到叶子节点时,将叶子节点的值存储为对应区间的值。

线段树查询
线段树查询可以通过递归的方式实现。对于一个查询区间[l, r],我们从根节点开始,根据区间与节点所代表区间的关系,逐步向下遍历子树,直到到达叶子节点。在遍历过程中,我们可以根据需要更新节点的值,例如查询最大值时,如果当前节点的值比查询区间的值更大,就将当前节点的值更新为查询区间的值。具体的查询过程如下:
1.从根节点开始,如果当前节点所代表的区间与查询区间有交集,就继续向下遍历。
2.如果当前节点的左儿子节点存在,并且左儿子节点所代表的区间与查询区间有交集,就递归查询左儿子节点。
3.如果当前节点的右

示例1:
Segment Tree 线段树算法(java)_第1张图片 线段树是一种二叉搜索树。他将一段区间划分为若干个单位区间,每个节点之间存储一个区间。思想类似于分治思想。

如图所示,线段树中每一个节点都存储着区间[1,10]中的信息,叶子节点的L = R。大致思想为:将大区间平分为2个小区间,每一个小区间再平分为更小的2个区间,以此类推直到每个区间的L = R,通过对这些区间的修改和查询来实现对大区间的修改和查询。
单点查找、修改的时间复杂度:O(log2n)
线段树维护的问题必须满足区间加法 例如:[1,3] + [2,4] = [1,4]。

代码演示

 public static class SegmentTree{
        //记录原数组的长度, 线段树下标是从1 开始,原数组是从0开始,
        private int MAXN;
        //保存原数组的信息,不过下标从1开始了
        private int[]arr;
        //模拟线段树维护记录区间和
        private int[]sum;
        //为累加和懒加载
        private int[]lazy;
        //区间更新的值
        private int[]change;
        //是否更新的标记
        private boolean[]update;

        public SegmentTree(int[] origin) {
            MAXN = origin.length + 1;
            for (int i = 1; i < MAXN;i++){
                arr[i] = origin[i - 1];
            }
            //用来记录逻辑概念中,某一范围的累加和信息。
            sum = new int[MAXN << 2];
            //用来支持逻辑概念中,某一个范围沒有往下透传的累加任务
            lazy = new int[MAXN << 2];
            //用来支持逻辑概念中,某一个范围更新任务,更新成了什么
            change = new int[MAXN << 2];
            // 用来支持逻辑概念中,某一个范围有没有更新操作的任务
            update = new boolean[MAXN << 2];
        }

        /**
         * rt 代表逻辑概念中 所在的位置
         * @param rt
         */
        public void pushUp(int rt){
            sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
        }

        /**
         * 初始化线段树
         * 初始化时 先把sum数组填好
         * @param l
         * @param r
         * @param rt
         */
        public void build(int l,int r,int rt){
            if (l == r){
                sum[rt] = arr[l];
                return;
            }
            int mid = (r + l) >> 1;
            build(l,mid,rt << 1);
            build(mid + 1,r,rt << 1 | 1);
            pushUp(rt);
        }
        /**
         * 任务往下分配
         * @param rt 当前位置
         * @param ln 分配任务的左边界
         * @param rn 分配任务的右边界
         */
        private void pushDown(int rt,int ln,int rn){
            if (update[rt]){
                update[rt << 1] = true;
                update[rt << 1 | 1] = true;
                change[rt << 1] = change[rt];
                change[rt << 1 | 1] = change[rt];
                lazy[rt << 1] = 0;
                lazy[rt << 1 | 1] = 0;
                sum[rt << 1] = change[rt] * ln;
                sum[rt << 1 | 1] = change[rt] * rn;
                update[rt] = false;
            }
            //懒加载的数据分发下去
            if (lazy[rt] != 0){
                lazy[rt << 1] += lazy[rt];
                sum[rt << 1] += lazy[rt] * ln;
                lazy[rt << 1 | 1] += lazy[rt];
                sum[rt << 1 | 1] += lazy[rt] * rn;
                lazy[rt] = 0;
            }
        }

        /**
         * L - R 任务的范围,C 修改的大小
         * l r ,rt 所负责的范围。
         * @param L
         * @param R
         * @param C
         * @param l
         * @param r
         * @param rt
         */
        public void add(int L,int R,int C,int l,int r,int rt){
            if (L <= l && r <= R){
                sum[rt] += C * (r - l + 1);
                lazy[rt] += C;
                return;
            }
            int mid = (r + l) >> 1;
            //任务没有全包,就下发。
            pushDown(rt,mid - l + 1,r - mid);
            if (L <= mid){
                add(L,R,C,l,mid,rt << 1);
            }
            if (R > mid){
                add(L,R,C,mid + 1,r,rt << 1 | 1);
            }
            pushUp(rt);
        }
       

        /**
         *  //L - R 范围内所有值都变成 C
         *  //l - r 是 rt 所在位置的包含数字的范围
         * @param L
         * @param R
         * @param C
         * @param l
         * @param r
         * @param rt
         */
        public void update(int L,int R,int C,int l,int r,int rt){
            //所在范围被全包了。
            if (L <= l && r <= R){
                change[rt] = C;
                update[rt] = true;
                sum[rt] = (r - l + 1) * C;
                lazy[rt] = 0;
                return ;
            }

            int mid = (r + l) >> 1;
            pushDown(rt,mid - l + 1,r - mid);
            if (L <= mid){
                update(L,R,C,l,mid ,rt << 1);
            }
            if (R > mid){
                update(L,R,C,mid + 1,r,rt << 1 | 1);
            }
            pushUp(rt);
        }

       

        /**
         * 查询
         *  //L - R 要查询的范围
         *   //l - r 是 rt 所包含的范围。
         * @param L
         * @param R
         * @param l
         * @param r
         * @param rt
         * @return
         */
        public long query(int L,int R,int l,int r,int rt){
            if (L <= l && r <= R){
                return sum[rt];
            }
            int mid = (r + l) >> 1;
            pushDown(rt,mid - l + 1,r - mid);
            long ans = 0;
            if (L <= mid){
                ans += query(L,R,l,mid,rt << 1);
            }
            if (R > mid){
                ans += query(L,R,mid + 1, r, rt << 1 | 1);
            }
            return ans;
        }
    }

蓄水池算法

蓄水池算法

你可能感兴趣的:(java,算法,数据结构,算法,java,开发语言,数据结构,hash-index)