LeetCode50天刷题计划第二季（Day 33 — 二叉搜索树迭代器（13.00-13.40）

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

二叉搜索树（二叉查找树）变二叉线索树（但不要next指针

一、题目

二叉搜索树迭代器

实现一个二叉搜索树迭代器类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 是树的高度。

二、思路

二叉树结点没有next属性，如果可以给结点加上一个next的属性就很方便了（二叉线索树），可以实现空间复杂度O(1)，但此处没有
采用中序遍历二叉树(左中右)，把每个结点保存在列表中，指针记录访问到的index，然后即可实现随机查询

三、代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
#         self.next = next 
class BSTIterator:
    def __init__(self, root: Optional[TreeNode]):
        self.root=TreeNode(-1,None,root) #建立空的根结点，值设置为-1，因此其右子树就是原先的二叉树
        self.pointer=0 #整个迭代器维护一个指针变量，指针初始时指向空根节点
        self.stack=list()
        #中序遍历函数
        def midTravel(root): 
            if(root == None):
                return 
            midTravel(root.left) #左子树
            self.stack.append(root) #入队
            midTravel(root.right) #右子树
        #调用函数
        midTravel(self.root)
        self.length=len(self.stack)
    
    def next(self) -> int:
        #队首元素出队
        self.pointer+=1#下一个元素
        return self.stack[self.pointer].val

    def hasNext(self) -> bool:
        #只要栈不空
        if(self.pointer != self.length-1):
            return True
        else:
            return False

# Your BSTIterator object will be instantiated and called as such:
# obj = BSTIterator(root)
# param_1 = obj.next()
# param_2 = obj.hasNext()