news 2026/4/18 7:38:01

TreeMap与TreeSet的实现原理

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
TreeMap与TreeSet的实现原理

JavaTreeMapTreeSet:基于红黑树的有序集合

TreeMapTreeSet是 Java 集合框架中的两种重要数据结构,它们都基于红黑树(Red-Black Tree)实现,提供了有序的元素存储和操作方式。TreeMap用于存储键值对(key-value),而TreeSet用于存储不重复的元素。它们的底层实现原理非常相似,但由于存储结构和用途的不同,具体实现也有所区别。

1. 红黑树数据结构的基础

红黑树概述

红黑树是一种自平衡的二叉查找树,具有以下特点:

  1. 每个节点要么是红色,要么是黑色。
  2. 根节点是黑色。
  3. 每个叶子节点(NIL 节点)是黑色。
  4. 如果一个红色节点的父节点是红色的,则违反红黑树规则,因此红色节点的子节点必须是黑色。
  5. 从根节点到每个叶子节点的路径上,黑色节点的数量必须相同。
  6. 红黑树的高度是平衡的,因此它的查找、插入和删除操作的时间复杂度为O(log n)

2.TreeMap的实现原理

TreeMap 基本结构

TreeMap实现了NavigableMap接口,而NavigableMap继承自SortedMapTreeMap的核心数据结构是一个红黑树。每个树节点包含一个键值对(key 和 value),并根据 key 进行排序。

TreeMap 插入过程

TreeMap中的插入操作通过红黑树实现,在插入新节点时会保持红黑树的平衡。插入的过程包括:

  1. 查找位置:根据 key 比较找到合适的位置。
  2. 插入节点:找到位置后创建新的节点,并插入到树中。
  3. 平衡调整:插入后调用fixAfterInsertion来确保红黑树的平衡性。
fixAfterInsertion:插入后平衡树
  1. 颜色调整:如果父节点是红色节点,需要进行颜色调整。
  2. 旋转调整:通过左旋或右旋来保持红黑树的平衡。

3.TreeSet的实现原理

TreeSetTreeMap非常相似,不同之处在于TreeSet不存储值,只存储键。内部结构使用TreeMap来实现,而TreeMap中的 key 就是TreeSet中的元素。

TreeSet 的实现

  • TreeSet底层通过TreeMap来存储元素,所有的操作(如添加、删除、查找等)都会通过TreeMap实现。
  • TreeMap存储元素的同时保证它们的有序性。

4. 查找与删除操作

查找操作

TreeMapTreeSet的查找操作都基于红黑树的查找方式。通过compare()方法来比较键值,查找时间复杂度是O(log n)

删除操作

删除操作需要先找到要删除的节点,然后通过旋转和颜色调整来保持红黑树的平衡。

示例代码:TreeMap的删除操作

java复制

public V remove(Object key) { Entry<K, V> p = getEntry(key); if (p == null) return null; deleteEntry(p); return p.value; } private void deleteEntry(Entry<K, V> p) { if (p.left != null && p.right != null) { Entry<K, V> s = successor(p); p.key = s.key; p.value = s.value; p = s; } Entry<K, V> replacement = (p.left != null ? p.left : p.right); if (replacement != null) { replacement.parent = p.parent; if (p.parent == null) root = replacement; else if (p == p.parent.left) p.parent.left = replacement; else p.parent.right = replacement; p.left = p.right = p.parent = null; } else if (p.parent == null) { root = null; } else { if (p == p.parent.left) p.parent.left = null; else p.parent.right = null; p.parent = null; } size--; modCount++; }

5. 总结

  • TreeMapTreeSet都基于红黑树实现,提供了高效的查找、插入和删除操作。
  • TreeMap存储键值对,TreeSet只存储键。
  • 插入和删除操作会触发红黑树的旋转和颜色调整,以保证树的平衡性。
  • TreeSetTreeMap的一个封装,利用TreeMap来实现元素的有序存储。
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/17 19:51:10

集合框架的设计思想与架构

Java集合框架&#xff1a;设计思想、实现原理与性能优化 Java集合框架是Java语言中用于存储和处理对象的工具集。它通过接口抽象和类的实现&#xff0c;提供了灵活、可扩展的集合操作方式。本文将从源码角度详细分析集合框架的设计思想、实现原理以及性能优化策略。 1. 集合框…

作者头像 李华
网站建设 2026/3/24 10:43:21

Live Avatar生成质量差?四大优化方法提升清晰度

Live Avatar生成质量差&#xff1f;四大优化方法提升清晰度 1. Live Avatar模型简介与硬件限制 Live Avatar是由阿里联合高校开源的数字人生成模型&#xff0c;专注于实时驱动的高质量视频生成。它融合了文本理解、语音驱动、图像建模与视频合成能力&#xff0c;支持从单张参…

作者头像 李华
网站建设 2026/3/14 4:02:30

如何统计GPEN处理成功率?日志分析与报表生成技巧

如何统计GPEN处理成功率&#xff1f;日志分析与报表生成技巧 1. 为什么需要统计处理成功率&#xff1f; 你可能已经用GPEN修复过几十张甚至上百张老照片&#xff0c;也经历过“点下按钮→等待→发现某几张没出来”的困惑。但你有没有想过&#xff1a;到底有多少张成功了&…

作者头像 李华
网站建设 2026/4/13 17:01:40

verl未来发展方向:路线图与生态展望

verl未来发展方向&#xff1a;路线图与生态展望 1. verl 是什么&#xff1f;一个为大模型后训练而生的强化学习框架 你可能已经听说过 RLHF&#xff08;基于人类反馈的强化学习&#xff09;&#xff0c;也用过类似 DeepSpeed-RLHF 的工具来微调大语言模型。但当你真正想把 RL…

作者头像 李华
网站建设 2026/4/18 3:52:42

加密数据看不懂?智能解密引擎让复杂算法无所遁形

加密数据看不懂&#xff1f;智能解密引擎让复杂算法无所遁形 【免费下载链接】help_tool 推理算法助手(降维打击) 项目地址: https://gitcode.com/gh_mirrors/he/help_tool 面对加密数据时&#xff0c;是否曾因算法未知而束手无策&#xff1f;本文介绍的智能解密工具通…

作者头像 李华