[ADT实践应用]Java中线段树求区间和的实现

这学期粗浅的学习了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的实现,理论上其它线段树可以维护的内容也都可以存储在这个类中,比如乘法运算标记,最大值标记等。但是我没有写

忽然发现回来写写博客其实也不错,以后有空就继续写写文章咯。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值