[Leetcode]31.二叉搜索树迭代器

2021年11月20日 阅读数:4
这篇文章主要向大家介绍[Leetcode]31.二叉搜索树迭代器,主要内容包括基础应用、实用技巧、原理机制等方面,希望对大家有所帮助。

题目:实现一个二叉搜索树迭代器类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 操做app

 

思想:中序遍历存入序列化器定义的数组,分别读取数组的后一个值和判空便可。函数

type BSTIterator struct {
	arr []int
}

func Constructor(root *TreeNode) BSTIterator {
	var bstierator BSTIterator
	if root == nil {
		return bstierator
	}
	var dfs func(root *TreeNode)
	dfs = func(root *TreeNode) {
		if root.Left != nil {
			dfs(root.Left)
		}
		bstierator.arr = append(bstierator.arr, root.Val)
		if root.Right != nil {
			dfs(root.Right)
		}
	}
	dfs(root)
	return bstierator
}

func (this *BSTIterator) Next() int {
		r := this.arr[0]
		this.arr = this.arr[1:]
		return r
}

func (this *BSTIterator) HasNext() bool {
	if len(this.arr) != 0 {
		return true
	}
	return false
}

题目来源:https://leetcode-cn.com/problems/binary-search-tree-iterator

this