【线段树】java代码实现

简介

线段树是指使用二叉树形式表示一串数组形式的值，其中数组形式不被修改，实现区域求和区域修改。

代码实现

节点类

构造属性

min和max表示线段的起点和终点，site表示该节点在线段中的我位置，value表示该节点被赋予的值，left和right表示该节点关联的左右子树

    int min=-1;
    int max=-1;
    int site=-1;
    int value=-1;
    Point left=null;
    Point right=null;

构造方法

构造方法接受数组和偏移量作为入参，默认偏移量为0。若数组已被拆解为一个单一元素，则创建这个元素节点。
若数组长度超过2则用偏移量作为min点，用偏移量+数组长度作为max点，表示该节点包含的节点范围。
然后将数组拆解为两个，前半部分传递至左子树，后半部分传递至右子树。子树构建完毕则递归子树值至目前节点，若子节点有值则直接传递子节点值，若子节点无值则向下递归取子节点的节点的值，线段树相较于数组的算法优化就体现在这里。

    Point(int[] arr,int padding){
        if(arr.length==1){
            this.value=arr[0];
            this.site=padding;
        }else{
            this.min = padding;
            this.max = padding+arr.length-1;
            int[] leftArr = Arrays.copyOfRange(arr,0,arr.length/2);
            int[] rightArr = Arrays.copyOfRange(arr,arr.length/2,arr.length);
            this.left = new Point(leftArr,padding);
            this.right = new Point(rightArr,padding+arr.length/2);
            int leftValue = this.left==null?0:this.left.getValue();
            int rightValue = this.right==null?0:this.right.getValue();
            this.value = leftValue+rightValue;
        }
    }
    private int getValue(){
        if(this.value!=-1){
            return this.value;
        }else{
            int leftValue = this.left==null?0:this.left.getValue();
            int rightValue = this.right==null?0:this.right.getValue();
            return leftValue+rightValue;
        }
    }

区域修改方法

区域修改方法，传入起始点和终点（begin，end），修改方式（这里仅做了加减两种，cum），修改值（区域加num或区域减num）
将起点和终点转为数组传递至edit方法，若数组长度为1则说明已传递至叶子节点，直接进行修改即可。
非叶子节点则需要处理数组。循环数组区分数组中的节点位置应传递至左/右节点，将拆分结束的节点传递至目标节点，同时更具当前数组长度修改当前节点值。

    public void update(int begin,int end,int cum,int num){
        int[] arr =getArr(begin,end);
        this.edit(arr,cum,num);
    }
    private void edit(int[] arr,int cum,int num){
        if(arr.length==1){
            if(cum==0){
                this.value=this.value+num;
            }
            if(cum==1){
                this.value=this.value-num;
            }
        }else{
            int x=0;
            int toLeft=0;
            int toRight=0;
            for(int i:arr){
                if(i>=this.min || i<=this.max){
                    x++;
                }
                if(this.left.min<=i && i<=this.left.max){
                    toLeft++;
                }
                if(this.right.min<=i && i<=this.right.max){
                    toRight++;
                }
            }
            if(cum==0){
                this.value =this.value + num*x;
            }
            if(cum==1){
                this.value =this.value - num*x;
            }
            int[] toLeftArr = Arrays.copyOfRange(arr,0,toLeft);
            int[] toRightArr = Arrays.copyOfRange(arr,arr.length-toRight,arr.length);
            if(this.left!=null){
                this.left.edit(toLeftArr,cum,num);
            }
            if(this.right!=null){
                this.right.edit(toRightArr,cum,num);
            }
        }
    }

区域查询方法

取起点和终点以递归方式传递值，若完全包含当前节点，则之间传递节点本身，若不完全包含则将方法向下传递

    public int get(int begin,int end){
        if(begin>end){
            return 0;
        }
        if((this.min==-1 && this.max==-1) || (this.min==begin && this.max==end)){
            return this.value;
        }
        if(this.left!=null && this.right!=null){
            return this.left.get(begin,this.left.max)+this.right.get(this.right.min,end);
        }
        return 0;
    }

show方法

先序递归，从根节点开始以二叉树形式展示整个线段树

    public void showTree(){
        if(this.min==-1 && this.max==-1){
            System.out.println(this.site+"------------------>"+this.value);
        }else{
            System.out.println(this.min+"~"+this.max+"-------------->"+this.value);
        }
        if(this.left!=null){
            this.left.showTree();
        }
        if(this.right!=null){
            this.right.showTree();
        }
    }

测试

构造长度为10的连续数组，依次执行构造，区域修改，区域查询

    public static void main(String[] args){
        int[] arr = {0,1,2,3,4,5,6,7,8,9};
        Point p = new Point(arr,0);
        p.showTree();
        p.update(2,6,0,1);
        p.showTree();
        System.out.println("区域查询"+p.get(5,9));
    }

0~9-------------->45
0~4-------------->10
0~1-------------->1
0------------------>0
1------------------>1
2~4-------------->9
2------------------>2
3~4-------------->7
3------------------>3
4------------------>4
5~9-------------->35
5~6-------------->11
5------------------>5
6------------------>6
7~9-------------->24
7------------------>7
8~9-------------->17
8------------------>8
9------------------>9
0~9-------------->50
0~4-------------->13
0~1-------------->1
0------------------>0
1------------------>1
2~4-------------->12
2------------------>2
3~4-------------->9
3------------------>3
4------------------>4
5~9-------------->37
5~6-------------->13
5------------------>5
6------------------>6
7~9-------------->24
7------------------>7
8~9-------------->17
8------------------>8
9------------------>9
区域查询37

完整代码

节点类

package LineTree;

import java.util.Arrays;

public class Point {

    int min=-1;
    int max=-1;
    int site=-1;
    int value=-1;
    Point left=null;
    Point right=null;

    Point(int[] arr,int padding){
        if(arr.length==1){
            this.value=arr[0];
            this.site=padding;
        }else{
            this.min = padding;
            this.max = padding+arr.length-1;
            int[] leftArr = Arrays.copyOfRange(arr,0,arr.length/2);
            int[] rightArr = Arrays.copyOfRange(arr,arr.length/2,arr.length);
            this.left = new Point(leftArr,padding);
            this.right = new Point(rightArr,padding+arr.length/2);
            int leftValue = this.left==null?0:this.left.getValue();
            int rightValue = this.right==null?0:this.right.getValue();
            this.value = leftValue+rightValue;
        }
    }
    private int getValue(){
        if(this.value!=-1){
            return this.value;
        }else{
            int leftValue = this.left==null?0:this.left.getValue();
            int rightValue = this.right==null?0:this.right.getValue();
            return leftValue+rightValue;
        }
    }
    private int[] getArr(int begin,int end){
        if(begin>end){
            System.out.println("数据错误");
            return null;
        }
        int length = end-begin;
        int[] arr = new int[length+1];
        for(int i=0;i<length+1;i++){
            arr[i] =begin;
            begin+=1;
        }
        return arr;
    }
    public void update(int begin,int end,int cum,int num){
        int[] arr =getArr(begin,end);
        this.edit(arr,cum,num);
    }
    private void edit(int[] arr,int cum,int num){
        if(arr.length==1){
            if(cum==0){
                this.value=this.value+num;
            }
            if(cum==1){
                this.value=this.value-num;
            }
        }else{
            int x=0;
            int toLeft=0;
            int toRight=0;
            for(int i:arr){
                if(i>=this.min || i<=this.max){
                    x++;
                }
                if(this.left.min<=i && i<=this.left.max){
                    toLeft++;
                }
                if(this.right.min<=i && i<=this.right.max){
                    toRight++;
                }
            }
            if(cum==0){
                this.value =this.value + num*x;
            }
            if(cum==1){
                this.value =this.value - num*x;
            }
            int[] toLeftArr = Arrays.copyOfRange(arr,0,toLeft);
            int[] toRightArr = Arrays.copyOfRange(arr,arr.length-toRight,arr.length);
            if(this.left!=null){
                this.left.edit(toLeftArr,cum,num);
            }
            if(this.right!=null){
                this.right.edit(toRightArr,cum,num);
            }
        }
    }

    public int get(int begin,int end){
        if(begin>end){
            return 0;
        }
        if((this.min==-1 && this.max==-1) || (this.min==begin && this.max==end)){
            return this.value;
        }
        if(this.left!=null && this.right!=null){
            return this.left.get(begin,this.left.max)+this.right.get(this.right.min,end);
        }
        return 0;
    }

    public void showTree(){
        if(this.min==-1 && this.max==-1){
            System.out.println(this.site+"------------------>"+this.value);
        }else{
            System.out.println(this.min+"~"+this.max+"-------------->"+this.value);
        }
        if(this.left!=null){
            this.left.showTree();
        }
        if(this.right!=null){
            this.right.showTree();
        }
    }
}

测试类

package LineTree;


public class testMain {

    public static void main(String[] args){
        int[] arr = {0,1,2,3,4,5,6,7,8,9};
        Point p = new Point(arr,0);
        p.showTree();
        p.update(2,6,0,1);
        p.showTree();
        System.out.println("区域查询"+p.get(5,9));
    }

}