这学期粗浅的学习了Java的语法,以及系统地学习了ADT和OOP的概念,心血来潮自己去弄了一个线段树的Java实现,顺便理解课程的ADT相关思想。
线段树应该不需要我去详细地说太多的内容,这也算是比较基础的一种数据结构了。既然是面向对象编程了,就要改一改以前竞赛乱写乱用全局变量的方法。但是由于线段树实现实在过于简单,所以我认为还是使用数组表示是最好的解决方案。我们直接通过ArrayList来存储前缀和以便预处理,同时使用tree数组来表示我们的线段树。
关于初始化大小可以证明只需要开出初始数组大小的四倍的空间即可
package p1;
import java.util.*;
public class Tree {
ArrayList<Node> tree=new ArrayList<>();
ArrayList<Integer> sum = new ArrayList<>();
private int n;
public Tree(ArrayList<Integer> arr) {
int size=arr.size(),now=0;
sum.add(0);
for(int i=0;i<size;++i) {
now+=arr.get(i);
sum.add(now);
}
for(int i=0;i<=size*4;++i) {
tree.add(new Node(0,i));
}
n=size;
build(1,size,1);
}
private void build(int l,int r,int nodeid) {
int mid=(l+r)/2;
if(l==r) {
tree.set(nodeid, new Node(sum.get(r)-sum.get(l-1),nodeid));
return;
}
else {
build(l,mid,nodeid*2);
build(mid+1,r,nodeid*2 + 1);
tree.set(nodeid, new Node(sum.get(r)-sum.get(l-1),nodeid));
}
return;
}
private void modify(int p,int val,int l,int r,int nodeid) {
if(l==r) {
tree.set(nodeid, new Node(val,nodeid));
return;
}
else {
int mid=(l+r)/2;
if(p<=mid) modify(p,val,l,mid,nodeid*2);
else modify(p,val,mid+1,r,nodeid*2+1);
tree.set(nodeid, new Node(tree.get(nodeid*2).sum+tree.get(nodeid*2+1).sum,nodeid));
}
return;
}
private int query(int ql,int qr,int l,int r,int nodeid) {
if(ql==l && qr==r) {
return tree.get(nodeid).sum;
}
else {
int mid=(l+r)/2;
if(qr<=mid) {
return query(ql,qr,l,mid,nodeid*2);
}
else if(ql>mid) {
return query(ql,qr,mid+1,r,nodeid*2+1);
}
else {
int res=0;
res=query(ql,mid,l,mid,nodeid*2)+query(mid+1,qr,mid+1,r,nodeid*2+1);
return res;
}
}
}
public int Q(int l,int r) {
return query(l,r,1,n,1);
}
public void M(int p,int val) {
modify(p,val,1,n,1);
return;
}
}
由于希望封装性能好一些,所以说就单纯使用了两个简单的函数将对应的两个函数保护起来,在外界传入变量的时候也不会令人困惑多出的需要变量是什么。
package p1;
public class Node {
int sum,id;
public Node(int sum,int id) {
this.sum=sum;
this.id=id;
}
}
上面的是Node的实现,理论上其它线段树可以维护的内容也都可以存储在这个类中,比如乘法运算标记,最大值标记等。但是我没有写
忽然发现回来写写博客其实也不错,以后有空就继续写写文章咯。