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

如何高效合并两个二叉搜索树的元素并按升序输出

轻芳酱_8905

轻芳酱_8905

发布时间:2026-08-07 09:24:21

|

273人浏览过

|

来源于php中文网

原创

如何高效合并两个二叉搜索树的元素并按升序输出

本文介绍在不额外排序的前提下,以 o(m+n) 时间复杂度合并两个 bst 的所有节点值并返回升序列表的最优解法,重点讲解双栈迭代模拟中序遍历+归并的思想与实现。

本文介绍在不额外排序的前提下,以 o(m+n) 时间复杂度合并两个 bst 的所有节点值并返回升序列表的最优解法,重点讲解双栈迭代模拟中序遍历+归并的思想与实现。

在处理两个二叉搜索树(BST)时,一个经典需求是:将两棵树的所有节点值合并,并以严格升序排列返回 List。由于 BST 的中序遍历天然产生升序序列,直观思路是分别中序遍历两棵树、合并两个有序列表——但若使用递归中序再排序(如 list.addAll(inorder1); list.addAll(inorder2); Collections.sort(list)),时间复杂度退化为 O((m+n) log(m+n)),且空间开销冗余。

更优解法应充分利用 BST 的有序性,避免显式排序。关键洞察在于:我们不需要预先生成完整中序序列,而只需“按需获取”当前最小未访问节点值——这正是归并两个有序流的核心思想。但由于递归难以同步控制两棵树的遍历进度(递归调用栈无法独立暂停/恢复两棵子树),纯递归方式无法直接实现“双指针式”的协同遍历。原因在于:单次递归调用只能深入一棵树的一个分支,返回时会同时回溯两棵树的调用栈,丧失对各自遍历状态的细粒度控制。

因此,工业级可靠解法采用双栈迭代模拟中序遍历 + 归并逻辑,时间复杂度 O(m+n),空间复杂度 O(h₁+h₂)(h₁、h₂ 为两树高度)。其核心逻辑如下:

  • 维护两个栈 stack1 和 stack2,分别模拟树1和树2的中序遍历过程;
  • 使用两个指针 node1 和 node2 指向当前待处理节点;
  • 循环中:先将各自左链全部压栈(即找到各自最左下节点);
  • 比较两栈顶元素(若某栈为空则取另一栈),弹出较小者,加入结果,并将其右子树作为新起点继续压栈;
  • 直到两栈均空且两指针均为 null。

以下是简洁、健壮的 Java 实现:

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()) {
        // 将 node1 左链全压入 stack1
        while (node1 != null) {
            stack1.push(node1);
            node1 = node1.left;
        }
        // 将 node2 左链全压入 stack2
        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;
}

⚠️ 注意事项:

  • 边界安全:每次比较前必须检查栈是否为空,避免 peek() 抛出 NoSuchElementException;
  • 稳定性保证:当两栈顶值相等时,优先取 stack1 的元素(
  • 空间优化:使用 ArrayDeque 替代 Stack(后者基于过时的 Vector,线程安全但性能差);
  • 不可递归化:试图设计 recurse(TreeNode r1, TreeNode r2, List res) 方法会陷入逻辑死锁——递归无法“暂停”一棵树等待另一棵树推进,本质是双状态机问题,必须依赖显式栈管理。

总结:该双栈归并法兼具时间最优性与代码可读性,是解决“合并两 BST 有序元素”问题的标准范式。它将中序遍历的“左-根-右”过程解耦为可控的压栈/弹栈操作,再结合归并思想,实现了真正意义上的在线(on-the-fly)、流式(streaming)有序合并。

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

热门AI工具

更多
豆包大模型

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

WorkBuddy

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

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

DeepSeek

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

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

Atoms
Atoms Hot

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

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

相关专题

更多
Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

20

2026.10.28

Vite 如何搭配 Vue.js
Vite 如何搭配 Vue.js

Vite搭配Vue.js专题汇集创建Vue 3项目、配置开发环境、路径别名、Vue插件、代理、多环境构建与调试的已审核文章,帮助读者完成从初始化到日常开发的配置。

0

2026.10.28

Vite 项目目录怎么看
Vite 项目目录怎么看

Vite项目目录专题汇集入口文件、src源码、public静态资源、路径别名、项目根目录和构建输出目录相关的已审核文章,帮助读者理解各文件夹的用途与引用方式。

0

2026.10.28

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

120

2026.10.22

Gradle Wrapper配置教程
Gradle Wrapper配置教程

本专题介绍Gradle Wrapper的基本作用,涵盖文件组成、生成方法、版本配置和任务调用,帮助你理解项目如何使用指定的Gradle版本,减少不同开发环境带来的构建差异。

60

2026.10.22

Golangci-lint检查器使用教程
Golangci-lint检查器使用教程

本专题整理Golangci-lint常用Linters的用途与配置方法,覆盖错误处理、无效代码、安全、复杂度、命名和依赖检查,帮助项目合理选规则。

60

2026.10.22

Gradle怎么构建Java项目
Gradle怎么构建Java项目

本专题介绍Gradle Java项目构建方法,涵盖Java插件配置、依赖声明、源码编译、测试执行和打包检查,帮助你理解各阶段的任务关系,逐步完成项目构建。

40

2026.10.22

PHP时间戳漏洞修复教程
PHP时间戳漏洞修复教程

本专题汇总PHP常见时间戳漏洞,包括2038年溢出、时区漂移、时间戳伪造、重放攻击、时间盲注与日志审计失效等,并给出升级64位环境、使用 DateTimeImmutable、统一UTC、严格超时校验、预处理绑定、hash_equals 比对等修复方案,帮助开发者快速排查和加固时间相关代码。

60

2026.10.22

Golangci-lint配置文件与规则设置
Golangci-lint配置文件与规则设置

本专题聚焦Golangci-lint YAML配置文件,讲清运行参数、规则启停、目录排除、问题过滤及严重级别设置,便于团队统一检查标准。

20

2026.10.22

热门下载

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

精品课程

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

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