news 2026/7/25 19:40:49

LC.173 | 二叉搜索树迭代器 | 树 | 中序展开/栈模拟

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
LC.173 | 二叉搜索树迭代器 | 树 | 中序展开/栈模拟

输入:BST 根节点root,构造BSTIterator

要求:
实现一个按中序遍历输出 BST 的迭代器:

  • next():返回下一个最小值
  • hasNext():是否还有下一个元素

输出:按题意实现类方法(next/hasNext)。


思路:

思路 A:中序遍历“展开成线性表”

核心就是一句话:

BST 中序遍历 = 递增序列
先把整棵树中序遍历一遍,把结果按顺序塞进链表/数组,然后迭代器只是在这个线性结构上移动指针。

  1. 构造时inorder(root),按中序顺序把每个节点值 append 到单链表尾部。
  2. cur指向链表头(第一个最小值)。
  3. next()返回cur->val并前进。
  4. hasNext()cur != nullptr

优点:

  • 写起来直观,next/hasNext都是 O(1)。

缺点:

  • 构造函数要遍历整棵树,时间 O(N)。
  • 额外存了 N 个节点值,空间 O(N)。
  • 题目进阶想要更省空间(典型答案是 O(H))。

思路 B:用栈模拟中序遍历(更优解的核心思想)

迭代器本质是:每次只走到“下一个该访问的中序节点”,不提前把整棵树铺开。

维护一个栈stk

  • 构造时:把root的整条左链压栈(走到最左)。
  • next()
    1. 弹出栈顶x(当前最小)
    2. 如果x有右子树,把x->right的整条左链压栈
    3. 返回x->val
  • hasNext():看栈空不空

复杂度:

  • 思路 A(暴力)

    • 构造:O(N)
    • next/hasNext:O(1)
    • 空间:O(N)
  • 思路 B(栈模拟)

    • 构造:O(H)
    • next:均摊 O(1)(每个节点最多入栈出栈一次)
    • 空间:O(H)

//思路A 暴力classBSTIterator{public:BSTIterator(TreeNode*root){dummy=newListNode(0);tail=dummy;inorder(root);cur=dummy->next;}intnext(){intval=cur->val;cur=cur->next;returnval;}boolhasNext(){returncur!=nullptr;}private:structListNode{intval;ListNode*next;ListNode(intx):val(x),next(nullptr){}};ListNode*dummy;ListNode*tail;ListNode*cur;voidinorder(TreeNode*node){if(node==nullptr)return;inorder(node->left);tail->next=newListNode(node->val);tail=tail->next;inorder(node->right);}};//思路B 栈模拟classBSTIterator{public:BSTIterator(TreeNode*root){pushLeftChain(root);}intnext(){TreeNode*node=st.top();st.pop();intret=node->val;// 下一批候选:node 的右子树的最左链if(node->right){pushLeftChain(node->right);}returnret;}boolhasNext(){return!st.empty();}private:stack<TreeNode*>st;voidpushLeftChain(TreeNode*node){while(node){st.push(node);node=node->left;}}};
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/25 15:32:42

Java基于springboot+vue的毕业生离校管理系统的设计与实现

前言 毕业生离校管理系统是一款专为教育机构设计的软件工具&#xff0c;目的是简化和自动化毕业生在完成学业后离开学校前所需完成的多项手续。通过使用毕业生离校管理系统&#xff0c;学校管理人员可以高效地跟踪每位毕业生的离校状态&#xff0c;确保所有必要的步骤都得到妥善…

作者头像 李华
网站建设 2026/7/22 2:57:26

OpenMV中HOG特征提取全面讲解

在OpenMV上玩转HOG特征&#xff1a;从原理到实战的完整指南你有没有遇到过这样的场景&#xff1f;想在一块小小的MCU上做物体识别&#xff0c;却发现模型太大跑不动、训练数据不够、功耗又高得离谱。这时候&#xff0c;与其硬上深度学习&#xff0c;不如回头看看那些“老派但靠…

作者头像 李华
网站建设 2026/7/22 11:35:42

8个AI论文生成平台测评,降重与写作功能深度解析

在众多AI论文工具中&#xff0c;选择一款适合自己需求的平台可能令人眼花缭乱。本文将对比8款热门工具&#xff0c;重点聚焦降重、降AIGC率、写论文等功能。工具排名基于实测数据和用户反馈&#xff0c;确保客观实用性。以下是简要排行表&#xff08;基于效率、准确性和易用性&…

作者头像 李华
网站建设 2026/7/25 6:23:26

8个AI论文改写工具评测,降重与写作功能全面分析

在众多AI论文工具中&#xff0c;选择一款适合自己需求的平台可能令人眼花缭乱。本文将对比8款热门工具&#xff0c;重点聚焦降重、降AIGC率、写论文等功能。工具排名基于实测数据和用户反馈&#xff0c;确保客观实用性。以下是简要排行表&#xff08;基于效率、准确性和易用性&…

作者头像 李华