题目描述(中等难度)

一个二叉查找树,实现一个迭代器。 依次返回树中最小的值,hasNext 返回树中是否还有未返回的元素。

思路分析

如果做过 108 题 和 ,这里看到二分查找树,一定会想到它的一个性质。「二分查找树的中序遍历依次得到的元素刚好是一个升序数组」,所以这道题无非就是把中序遍历的结果依次输出即可。

至于中序遍历,在 94 题 已经做过了,里边介绍了三种解法,下边的解法也是依赖于之前中序遍历的代码,大家可以先过去看一下。

解法一

先不考虑题目 Note 中要求的空间复杂度和时间复杂度,简单粗暴一些。在构造函数中,对二叉树进行中序遍历,把结果保存到一个队列中,然后 next 方法直接执行出队操作即可。至于 hasNext 方法的话,判断队列是否为空即可。

空间复杂度,用队列保存了所有的节点值,所以是 O(n),此外中序遍历递归压栈的过程也需要 O(h) 的空间。

解法二

解法一中我们把所有节点都保存了起来,其实没必要一次性保存所有节点,而是需要一个输出一个即可。

所以我们要控制中序遍历的进程,不要让它一次性结束,如果用解法一递归的方法去遍历那就很难控制了,所以自然而然的会想到用栈模拟递归的过程。

下边是 解法二的代码。

和这道题糅合一起也很简单了,只需要把 stackcur 作为成员变量,然后每次调用 next 就执行一次 循环,并且要记录当前值,结束掉本次循环。

空间复杂度,这里只需要消耗栈的空间,也就是 O(h)

这道题关键就是要知道二叉搜寻树的中序遍历是升序序列,然后问题其实就转移到怎么进行中序遍历了。

添加好友一起进步~

如果觉得有帮助的话,可以点击 这里 给一个 star 哦 ^^