讲师中心 微信公众号
AI工具推荐 视频效率加速

合并两个二叉搜索树并按升序返回所有元素的高效实现

星明吖_9662

星明吖_9662

发布时间:2026-08-06 21:20:07

|

956人浏览过

|

来源于php中文网

原创

合并两个二叉搜索树并按升序返回所有元素的高效实现

本文详解如何在不额外排序的前提下,将两棵bst的所有节点值合并为一个升序列表,重点介绍基于双栈迭代的o(m+n)时间复杂度解法,并对比分析递归同步遍历的不可行性。

本文详解如何在不额外排序的前提下,将两棵bst的所有节点值合并为一个升序列表,重点介绍基于双栈迭代的o(m+n)时间复杂度解法,并对比分析递归同步遍历的不可行性。

二叉搜索树(BST)的核心性质是:中序遍历结果天然有序。因此,合并两个BST的升序结果,本质上等价于合并两个已排序序列——这正是归并排序中“归并”步骤的经典问题。但关键挑战在于:我们无法预先获取完整中序序列(否则需O(m+n)空间存储),也不能用简单递归同时驱动两棵树的深度优先遍历。

❌ 为什么不能用纯递归同步遍历?

直觉上,可能设想定义一个递归函数 recurse(TreeNode r1, TreeNode r2, List<Integer> res),根据 r1.val 和 r2.val 的大小关系决定向哪棵树深入。但这是不可行的:

  • 递归调用栈是单线程的,一次 return 只能回退一棵树的状态,无法独立维护两棵树各自的“当前最左未访问节点”;
  • BST 中序依赖隐式栈(调用栈或显式栈)来记住回溯路径,而双树递归会混淆两者的上下文,导致指针丢失或重复访问;
  • 不存在一种递归结构能保证“每次只推进较小值对应树的中序进度”,而不破坏另一棵树的状态一致性。

因此,任何试图用单一递归函数同步控制两棵BST遍历的方案,都会在边界情况(如某棵树提前遍历完、空子树、值相等等)下逻辑崩溃或难以维护状态。

✅ 正确解法:双栈模拟双指针中序迭代

最优解是用两个显式栈分别模拟两棵BST的中序迭代器,结合类似归并的逻辑逐个提取最小值:

public List<Integer> getAllElements(TreeNode root1, TreeNode root2) {
    List<Integer> result = new ArrayList<>();
    Deque<TreeNode> stack1 = new ArrayDeque<>();
    Deque<TreeNode> stack2 = new ArrayDeque<>();

    TreeNode node1 = root1, node2 = root2;

    while (node1 != null || !stack1.isEmpty() || node2 != null || !stack2.isEmpty()) {
        // 向左到底,压入所有左链节点(模拟中序的“找最左”)
        while (node1 != null) {
            stack1.push(node1);
            node1 = node1.left;
        }
        while (node2 != null) {
            stack2.push(node2);
            node2 = node2.left;
        }

        // 比较栈顶(即当前可得的最小候选值),取较小者
        if (stack2.isEmpty() || (!stack1.isEmpty() && stack1.peek().val <= stack2.peek().val)) {
            TreeNode curr = stack1.pop();
            result.add(curr.val);
            node1 = curr.right; // 推进到右子树,继续中序
        } else {
            TreeNode curr = stack2.pop();
            result.add(curr.val);
            node2 = curr.right;
        }
    }
    return result;
}

✅ 时间与空间复杂度

  • 时间复杂度:O(m + n) —— 每个节点最多入栈、出栈各一次;
  • 空间复杂度:O(h₁ + h₂) —— 栈深取决于两棵树的最大高度,远优于全量存储中序序列的 O(m + n) 空间(方法1)。

⚠️ 关键注意事项

  • 判空逻辑必须严谨:stack2.isEmpty() 优先判断,避免对空栈调用 peek();
  • <= 而非 < 保证相等值时优先取 root1 的元素,符合稳定归并语义;
  • node1/node2 在每次循环开始时被重置为 null,确保新左链扫描从头开始;
  • 该解法天然支持任意数量BST扩展(只需增加栈和节点变量)。

? 对比其他方法

方法 时间复杂度 空间复杂度 是否推荐 说明
两遍中序 + 排序 O((m+n) log(m+n)) O(m+n) ❌ 浪费BST有序性,排序冗余
自定义双迭代器类(如原题MyIterator) O(m+n) O(h₁+h₂) ✅ 封装性好,但代码量大
双栈归并(本文推荐) O(m+n) O(h₁+h₂) ✅✅✅ 简洁、高效、无额外类、易理解

综上,面对“合并多棵BST有序序列”的需求,应放弃同步递归幻想,坚定采用双栈迭代+归并逻辑这一经典范式——它既尊重BST的数学本质,又具备工程落地所需的简洁性与鲁棒性。

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

火山引擎

火山引擎是一款面向企业的云计算与AI服务平台。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

相关专题

更多
java
java

Java是一个通用术语,用于表示Java软件及其组件,包括“Java运行时环境 (JRE)”、“Java虚拟机 (JVM)”以及“插件”。php中文网还为大家带了Java相关下载资源、相关课程以及相关文章等内容,供大家免费下载使用。

9157

2023.06.15

java正则表达式语法
java正则表达式语法

java正则表达式语法是一种模式匹配工具,它非常有用,可以在处理文本和字符串时快速地查找、替换、验证和提取特定的模式和数据。本专题提供java正则表达式语法的相关文章、下载和专题,供大家免费下载体验。

6322

2023.07.05

java自学难吗
java自学难吗

Java自学并不难。Java语言相对于其他一些编程语言而言,有着较为简洁和易读的语法,本专题为大家提供java自学难吗相关的文章,大家可以免费体验。

5652

2023.07.31

java配置jdk环境变量
java配置jdk环境变量

Java是一种广泛使用的高级编程语言,用于开发各种类型的应用程序。为了能够在计算机上正确运行和编译Java代码,需要正确配置Java Development Kit(JDK)环境变量。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

1004

2023.08.01

java保留两位小数
java保留两位小数

Java是一种广泛应用于编程领域的高级编程语言。在Java中,保留两位小数是指在进行数值计算或输出时,限制小数部分只有两位有效数字,并将多余的位数进行四舍五入或截取。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

848

2023.08.02

java基本数据类型
java基本数据类型

java基本数据类型有:1、byte;2、short;3、int;4、long;5、float;6、double;7、char;8、boolean。本专题为大家提供java基本数据类型的相关的文章、下载、课程内容,供大家免费下载体验。

1196

2023.08.02

java有什么用
java有什么用

java可以开发应用程序、移动应用、Web应用、企业级应用、嵌入式系统等方面。本专题为大家提供java有什么用的相关的文章、下载、课程内容,供大家免费下载体验。

2409

2023.08.02

java在线网站
java在线网站

Java在线网站是指提供Java编程学习、实践和交流平台的网络服务。近年来,随着Java语言在软件开发领域的广泛应用,越来越多的人对Java编程感兴趣,并希望能够通过在线网站来学习和提高自己的Java编程技能。php中文网给大家带来了相关的视频、教程以及文章,欢迎大家前来学习阅读和下载。

19751

2023.08.03

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

80

2026.09.23

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn