news 2026/3/9 13:42:17

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/3/9 6:47:07

9种情绪精准识别!Emotion2Vec+系统实际应用案例展示

9种情绪精准识别&#xff01;Emotion2Vec系统实际应用案例展示 1. 这不是“能识别情绪”的演示&#xff0c;而是“真正在用情绪识别”的现场 你有没有遇到过这样的场景&#xff1a;客服团队每天处理上千通电话&#xff0c;却没人知道哪通电话里客户已经快被气哭了&#xff1b…

作者头像 李华
网站建设 2026/3/2 18:09:53

智能股票分析平台TradingAgents-CN部署指南:从入门到精通

智能股票分析平台TradingAgents-CN部署指南&#xff1a;从入门到精通 【免费下载链接】TradingAgents-CN 基于多智能体LLM的中文金融交易框架 - TradingAgents中文增强版 项目地址: https://gitcode.com/GitHub_Trending/tr/TradingAgents-CN 您是否曾经因复杂的金融数据…

作者头像 李华
网站建设 2026/3/8 22:43:23

3步掌握!新蜂商城API架构解密与实战

3步掌握&#xff01;新蜂商城API架构解密与实战 【免费下载链接】newbee-mall-api &#x1f525; &#x1f389;新蜂商城前后端分离版本-后端API源码 项目地址: https://gitcode.com/gh_mirrors/ne/newbee-mall-api 新蜂商城API是一套基于Spring Boot构建的电商后端系统…

作者头像 李华
网站建设 2026/3/5 0:42:08

一分钟了解Qwen3-Embedding-0.6B能做什么

一分钟了解Qwen3-Embedding-0.6B能做什么 你是否遇到过这些场景&#xff1a; 搜索一个技术问题&#xff0c;返回的文档里混着大量无关内容&#xff1b;客服知识库有上千条问答&#xff0c;但用户问“怎么重置密码”&#xff0c;系统却推荐了“如何修改绑定手机”&#xff1b;…

作者头像 李华
网站建设 2026/3/5 18:07:24

Unity性能调优实战:Lua脚本分析与游戏流畅度提升指南

Unity性能调优实战&#xff1a;Lua脚本分析与游戏流畅度提升指南 【免费下载链接】Miku-LuaProfiler 项目地址: https://gitcode.com/gh_mirrors/mi/Miku-LuaProfiler 在Unity开发中&#xff0c;Lua脚本的性能问题常常成为游戏流畅度的隐形杀手。当玩家抱怨卡顿、掉帧&…

作者头像 李华