21-3-28 力扣每日刷题 173. 二叉搜索树迭代器

最新推荐文章于 2024-03-20 05:30:00 发布

Hars、Wlgb

最新推荐文章于 2024-03-20 05:30:00 发布

阅读量93

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/Harswlgb/article/details/115278693

版权

算法专栏收录该内容

279 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何实现一个二叉搜索树的迭代器，该迭代器按照中序遍历顺序返回BST中的元素。通过使用栈来辅助，实现了next()和hasNext()操作的平均时间复杂度为O(1)，空间复杂度为O(h)，其中h是树的高度。博客详细阐述了迭代中序遍历的逻辑，包括将左子树压栈、处理当前节点和处理右子树的过程。

摘要由CSDN通过智能技术生成

二叉搜索树迭代器

实现一个二叉搜索树迭代器类BSTIterator ，表示一个按中序遍历二叉搜索树（BST）的迭代器：

BSTIterator(TreeNode root) 初始化 BSTIterator 类的一个对象。BST 的根节点 root 会作为构造函数的一部分给出。指针应初始化为一个不存在于 BST 中的数字，且该数字小于 BST 中的任何元素。
boolean hasNext() 如果向指针右侧遍历存在数字，则返回 true ；否则返回 false 。
int next()将指针向右移动，然后返回指针处的数字。

注意，指针初始化为一个不存在于 BST 中的数字，所以对 next() 的首次调用将返回 BST 中的最小元素。

你可以假设 next() 调用总是有效的，也就是说，当调用 next() 时，BST 的中序遍历中至少存在一个下一个数字。

示例：

在这里插入图片描述

输入
["BSTIterator", "next", "next", "hasNext", "next", "hasNext", "next", "hasNext", "next", "hasNext"]
[[[7, 3, 15, null, null, 9, 20]], [], [], [], [], [], [], [], [], []]
输出
[null, 3, 7, true, 9, true, 15, true, 20, false]
解释
BSTIterator bSTIterator = new BSTIterator([7, 3, 15, null, null, 9, 20]);
bSTIterator.next();    // 返回 3
bSTIterator.next();    // 返回 7
bSTIterator.hasNext(); // 返回 True
bSTIterator.next();    // 返回 9
bSTIterator.hasNext(); // 返回 True
bSTIterator.next();    // 返回 15
bSTIterator.hasNext(); // 返回 True
bSTIterator.next();    // 返回 20
bSTIterator.hasNext(); // 返回 False

提示：

树中节点的数目在范围 [1, 105] 内
0 <= Node.val <= 106
最多调用 105 次 hasNext 和 next 操作

进阶：

你可以设计一个满足下述条件的解决方案吗？next() 和 hasNext() 操作均摊时间复杂度为 O(1) ，并使用 O(h) 内存。其中 h 是树的高度。

package isLeetcode;

import java.util.ArrayDeque;
import java.util.Deque;

public class BSTIterator328 {
     public class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode() {}
      TreeNode(int val) { this.val = val; }
      TreeNode(int val, TreeNode left, TreeNode right) {
          this.val = val;
          this.left = left;
          this.right = right;
      }
    }

    /**
     * 中序遍历的基本逻辑是：处理左子树 -> 处理当前节点 -> 处理右子树。
     * 其中迭代做法是利用「栈」进行处理：
     * 1.先将当前节点的所有左子树压入栈，压到没有为止
     * 2.将最后一个压入的节点弹出（栈顶元素），加入答案
     * 3.将当前弹出的节点作为当前节点，重复步骤一
     */
    Deque<TreeNode> d = new ArrayDeque<>();
    public BSTIterator328(TreeNode root) {  //初始化操作，
        // 步骤 1
        dfsLeft(root);
    }

    public int next() {
        // 步骤 2
        TreeNode root =d.pollLast();
        int res=root.val;
        // 步骤 3
        root=root.right;
        // 步骤 1
        dfsLeft(root);
        return  res;
    }

    public boolean hasNext() {
        return  !d.isEmpty();
    }

    public void dfsLeft(TreeNode root){//将当前根节点的所以左节点压入栈
         while (root!=null){
             d.addLast(root);
             root=root.left;
         }
    }
}