Java数据结构之线段树详解
更新时间：2022年01月23日 10:44:00 作者：strongmore
线段树是一种二叉搜索树，与区间树相似，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。本文将介绍线段树的Java实现代码，需要的可以参考一下
介绍
线段树(又名区间树)也是一种二叉树，每个节点的值等于左右孩子节点值的和，线段树示例图如下
以求和为例，根节点表示区间0-5的和，左孩子表示区间0-2的和，右孩子表示区间3-5的和，依次类推。
代码实现
/**
* 使用数组实现线段树
*/
public class SegmentTree<E> {
private Node[] data;
private int size;
private Merger<E> merger;
public SegmentTree(E[] source, Merger<E> merger) {
this.merger = merger;
this.size = source.length;
this.data = new Node[size * 4];
buildTree(0, source, 0, size - 1);
}
public E search(int queryLeft, int queryRight) {
if (queryLeft < 0 || queryLeft > size || queryRight < 0 || queryRight > size
|| queryLeft > queryRight) {
throw new IllegalArgumentException("index is illegal");
}
return search(0, queryLeft, queryRight);
}
/**
* 查询区间queryLeft-queryRight的值
*/
private E search(int treeIndex, int queryLeft, int queryRight) {
Node treeNode = data[treeIndex];
int left = treeNode.left;
int right = treeNode.right;
if (left == queryLeft && right == queryRight) {
return elementData(treeIndex);
}
int leftTreeIndex = leftChild(treeIndex);
int rightTreeIndex = rightChild(treeIndex);
int middle = left + ((right - left) >> 1);
if (queryLeft > middle) {
return search(rightTreeIndex, queryLeft, queryRight);
} else if (queryRight <= middle) {
return search(leftTreeIndex, queryLeft, queryRight);
}
E leftEle = search(leftTreeIndex, queryLeft, middle);
E rightEle = search(rightTreeIndex, middle + 1, queryRight);
return merger.merge(leftEle, rightEle);
}
public void update(int index, E e) {
update(0, index, e);
}
/**
* 更新索引为index的值为e
*/
private void update(int treeIndex, int index, E e) {
Node treeNode = data[treeIndex];
int left = treeNode.left;
int right = treeNode.right;
if (left == right) {
treeNode.data = e;
return;
}
int leftTreeIndex = leftChild(treeIndex);
int rightTreeIndex = rightChild(treeIndex);
int middle = left + ((right - left) >> 1);
if (index > middle) {
update(rightTreeIndex, index, e);
} else {
update(leftTreeIndex, index, e);
}
treeNode.data = merger.merge(elementData(leftTreeIndex), elementData(rightTreeIndex));
}
private void buildTree(int treeIndex, E[] source, int left, int right) {
if (left == right) {
data[treeIndex] = new Node<>(source[left], left, right);
return;
}
int leftTreeIndex = leftChild(treeIndex);
int rightTreeIndex = rightChild(treeIndex);
int middle = left + ((right - left) >> 1);
buildTree(leftTreeIndex, source, left, middle);
buildTree(rightTreeIndex, source, middle + 1, right);
E treeData = merger.merge(elementData(leftTreeIndex), elementData(rightTreeIndex));
data[treeIndex] = new Node<>(treeData, left, right);
}
@Override
public String toString() {
return Arrays.toString(data);
}
private E elementData(int index) {
return (E) data[index].data;
}
private int leftChild(int index) {
return index * 2 + 1;
}
private int rightChild(int index) {
return index * 2 + 2;
}
private static class Node<E> {
E data;
int left;
int right;
Node(E data, int left, int right) {
this.data = data;
this.left = left;
this.right = right;
}
@Override
public String toString() {
return String.valueOf(data);
}
}
public interface Merger<E> {
E merge(E e1, E e2);
}
}
我们以LeetCode上的一个问题来分析线段树的构建，查询和更新，LeetCode307问题如下：
给定一个整数数组，查询索引区间[i,j]的元素的总和。
线段树构建
private void buildTree(int treeIndex, E[] source, int left, int right) {
if (left == right) {
data[treeIndex] = new Node<>(source[left], left, right);
return;
}
int leftTreeIndex = leftChild(treeIndex);
int rightTreeIndex = rightChild(treeIndex);
int middle = left + ((right - left) >> 1);
buildTree(leftTreeIndex, source, left, middle);
buildTree(rightTreeIndex, source, middle + 1, right);
E treeData = merger.merge(elementData(leftTreeIndex), elementData(rightTreeIndex));
data[treeIndex] = new Node<>(treeData, left, right);
}
测试代码
public class Main {
public static void main(String[] args) {
Integer[] nums = {-2, 0, 3, -5, 2, -1};
SegmentTree<Integer> segmentTree = new SegmentTree<>(nums, Integer::sum);
System.out.println(segmentTree);
}
}
最后构造出的线段树如下，前面为元素值，括号中为包含的区间。
递归构造过程为
- 当左指针和右指针相等时，表示为叶子节点
- 将左孩子和右孩子值相加，构造当前节点，依次类推
区间查询
/**
* 查询区间queryLeft-queryRight的值
*/
private E search(int treeIndex, int queryLeft, int queryRight) {
Node treeNode = data[treeIndex];
int left = treeNode.left;
int right = treeNode.right;
if (left == queryLeft && right == queryRight) {
return elementData(treeIndex);
}
int leftTreeIndex = leftChild(treeIndex);
int rightTreeIndex = rightChild(treeIndex);
int middle = left + ((right - left) >> 1);
if (queryLeft > middle) {
return search(rightTreeIndex, queryLeft, queryRight);
} else if (queryRight <= middle) {
return search(leftTreeIndex, queryLeft, queryRight);
}
E leftEle = search(leftTreeIndex, queryLeft, middle);
E rightEle = search(rightTreeIndex, middle + 1, queryRight);
return merger.merge(leftEle, rightEle);
}
查询区间2-5的和
public class Main {
public static void main(String[] args) {
Integer[] nums = {-2, 0, 3, -5, 2, -1};
SegmentTree<Integer> segmentTree = new SegmentTree<>(nums, Integer::sum);
System.out.println(segmentTree);
System.out.println(segmentTree.search(2, 5)); // -1
}
}
查询过程为
- 待查询的区间和当前节点的区间相等，返回当前节点值
- 待查询左区间大于中间区间值，查询右孩子
- 待查询右区间小于中间区间值，查询左孩子
- 待查询左区间在左孩子，右区间在右孩子，两边查询结果相加
更新
/**
* 更新索引为index的值为e
*/
private void update(int treeIndex, int index, E e) {
Node treeNode = data[treeIndex];
int left = treeNode.left;
int right = treeNode.right;
if (left == right) {
treeNode.data = e;
return;
}
int leftTreeIndex = leftChild(treeIndex);
int rightTreeIndex = rightChild(treeIndex);
int middle = left + ((right - left) >> 1);
if (index > middle) {
update(rightTreeIndex, index, e);
} else {
update(leftTreeIndex, index, e);
}
treeNode.data = merger.merge(elementData(leftTreeIndex), elementData(rightTreeIndex));
}
更新只影响元素值，不影响元素区间。
更新其实和构建的逻辑类似，找到待更新的实际索引，依次更新父节点的值。
总结
线段树可以很好地处理区间问题，如区间求和，求最大最小值等。
以上就是Java 数据结构之线段树详解的详细内容，更多关于Java 线段树的资料请关注脚本之家其它相关文章！
相关文章
下面小编就为大家带来一篇java异常处理的简单练习。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2016-06-06
Java修改eclipse中web项目的server部署路径问题
这篇文章主要介绍了Java修改eclipse中web项目的server部署路径,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-11-11
这篇文章主要介绍了Java二分查找算法实现代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-11-11
获取系统参数System.getProperties()与配置文件参数@Value(“${key}“)
这篇文章主要介绍了获取系统参数System.getProperties()与配置文件参数@Value("${key}"),本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-05-05
泛型在java中有很重要的地位，在面向对象编程及各种设计模式中有非常广泛的应用。java泛型知识点也是Java开发岗位必问的一个话题，今天小编就给大家普及下Java泛型常见面试题，感兴趣的朋友一起看看吧2021-06-06
这篇文章主要介绍了Spring中使用ehcache缓存的方法及原理详解,ehcache具有很强的灵活性,提供了LRU、LFU和FIFO缓存淘汰算法,Ehcache 1.2引入了最近最少使用、最久未使用和先进先 出缓存淘汰算法, 构成了完整的缓存淘汰算法,,需要的朋友可以参考下2024-01-01
详解SpringBoot中的index首页的访问、自定义Favicon图标
这篇文章主要介绍了SpringBoot中的index首页的访问、自定义Favicon图标,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2023-08-08
Redisson分布式锁 之前的基于注解的锁有一种锁是基本redis的分布式锁，这篇文章主要介绍了Java使用Redisson分布式锁实现原理，非常具有实用价值，需要的朋友可以参考下2018-10-10
这篇文章主要为大家介绍了ReentrantLock获取锁释放锁的流程示例分析详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-11-11
这篇文章主要为大家详细介绍了spring消息转换器的使用，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-07-07
最新评论