news 2026/6/25 19:59:46

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/6/25 17:55:48

Z-Image-Turbo影视概念设计:场景图生成系统搭建实战

Z-Image-Turbo影视概念设计&#xff1a;场景图生成系统搭建实战 1. 为什么影视概念设计师需要Z-Image-Turbo 你有没有遇到过这样的情况&#xff1a;客户凌晨两点发来需求——“明天上午十点前要三张赛博朋克风格的未来城市主视觉”&#xff0c;而你刚打开Photoshop&#xff0…

作者头像 李华
网站建设 2026/6/18 15:03:19

YOLOv11如何提升吞吐量?批量推理优化教程

YOLOv11如何提升吞吐量&#xff1f;批量推理优化教程 YOLOv11并不是官方发布的模型版本——当前YOLO系列最新稳定公开版本为YOLOv8&#xff08;Ultralytics官方维护&#xff09;与YOLOv10&#xff08;由清华大学团队于2024年提出&#xff09;。所谓“YOLO11”在主流开源社区、…

作者头像 李华
网站建设 2026/6/25 8:37:12

Glyph降本部署实战:单卡4090D运行,GPU费用省60%

Glyph降本部署实战&#xff1a;单卡4090D运行&#xff0c;GPU费用省60% 你是不是也遇到过这样的问题&#xff1a;想跑一个视觉推理模型&#xff0c;但动辄需要A100或H100双卡起步&#xff0c;光是云上租卡一个月就要好几千&#xff1f;推理速度慢、显存爆满、部署流程复杂………

作者头像 李华
网站建设 2026/6/13 8:28:16

解锁Windows效率:5个让经典界面重生的实用技巧

解锁Windows效率&#xff1a;5个让经典界面重生的实用技巧 【免费下载链接】Open-Shell-Menu 项目地址: https://gitcode.com/gh_mirrors/op/Open-Shell-Menu Windows界面优化是提升工作效率的关键&#xff0c;而经典开始菜单作为高效操作的核心&#xff0c;却在现代系…

作者头像 李华
网站建设 2026/6/19 11:08:19

戴森球工厂优化方案:从瓶颈诊断到物流效率提升的系统解决策略

戴森球工厂优化方案&#xff1a;从瓶颈诊断到物流效率提升的系统解决策略 【免费下载链接】FactoryBluePrints 游戏戴森球计划的**工厂**蓝图仓库 项目地址: https://gitcode.com/GitHub_Trending/fa/FactoryBluePrints 在戴森球计划的工厂建设过程中&#xff0c;许多玩…

作者头像 李华