LeetCode 题解 -173.二叉搜索树迭代器

最新推荐文章于 2021-03-28 18:01:30 发布

艾希逐月

最新推荐文章于 2021-03-28 18:01:30 发布

阅读量350

点赞数

分类专栏： LeetCode 算法文章标签：算法 leetcode

本文链接：https://blog.csdn.net/Ahri_J/article/details/106265889

版权

LeetCode 同时被 2 个专栏收录

18 篇文章 0 订阅

订阅专栏

算法

18 篇文章 0 订阅

订阅专栏

Leetcode 第 173. Binary Search Tree Iterator 题，题目难度 Medium。

一. 题目要求

实现一个二叉搜索树迭代器。你将使用二叉搜索树的根节点初始化迭代器。

调用 next() 将返回二叉搜索树中的下一个最小的数。

示例：

在这里插入图片描述


BSTIterator iterator = new BSTIterator(root);
iterator.next();    // 返回 3
iterator.next();    // 返回 7
iterator.hasNext(); // 返回 true
iterator.next();    // 返回 9
iterator.hasNext(); // 返回 true
iterator.next();    // 返回 15
iterator.hasNext(); // 返回 true
iterator.next();    // 返回 20
iterator.hasNext(); // 返回 false

next() 和 hasNext() 操作的时间复杂度是 O(1)，并使用 O(h) 内存，其中 h 是树的高度。

二. 解题思路 & 代码

本题考察的是对二叉搜索树的遍历，最重要的一点就是明白二叉搜索树的特点，这里引用极客时间《数据结构与算法之美》中的表述：

在树中的任意一个节点，其左子树中的每个节点的值，都要小于这个节点的值，而右子树节点的值都大于这个节点的值

题目要求我们的迭代器每次迭代返回的都是二叉树中的最小值，对于二叉搜索树可以采用中序遍历的方式将元素从小到大遍历一遍。

解法 1. 中序遍历 & List 迭代

每次 next 取值和 hasNext 判断没法直接通过操作二叉树来实现，因此需要进行数据结构的转换，这里我们使用 ArrayList，通过中序遍历将二叉树的值以从小到大的顺序存到 List 中，然后接下来的迭代针对 List 操作就非常方便了。

实现代码如下：

class BSTIterator {

    private List<TreeNode> nodeList = new ArrayList<>();
    private int len = 0;
    private int index = 0;

    public BSTIterator(TreeNode root) {
        this.inOrder(root);
        len = nodeList.size();
    }
	// 中序遍历
    private void inOrder(TreeNode node) {
        if (node == null) {
            return;
        }

        inOrder(node.left);
        nodeList.add(node);
        inOrder(node.right);
    }


    public int next() {
        return nodeList.get(index ++).val;
    }

    public boolean hasNext() {
        return index < len;
    }
}

上面代码实际运行效率为 15ms 左右，beats 90.49 % of java。

解法 2. 中序遍历 & 二叉树构建

上面的解法运行效率已经算不错了，但是有一个还可以改进的地方：

List 需要占用额外的存储空间，空间复杂度为 O(N)，不符合题目的要求。

为了不占用额外的存储空间，就只能在二叉树本身进行操作。回到二叉搜索树的特点，如果我们二叉搜索树进行如下变动，则二叉树就变成了类似链表的结构，
此时遍历就非常方便了。

在这里插入图片描述

上述变换核心逻辑是：

父节点与左子节点互换，父节点变为左子节点的左子节点
父节点与右子节点互换，右子节点变为其原父节点的父节点

实现代码如下：

public class BSTIterator {
    
    TreeNode current;
    public BSTIterator(TreeNode root) {
        current = root;
        if (root != null) {
        	// 创建临时根节点，其 left 节点即为二叉搜索树中的最小左子节点
            TreeNode head = new TreeNode(0);
            TreeNode lastOne = build(head, root);
            lastOne.left = null;
            // 从 最小左子节点开始遍历
            current = head.left;
        }
    }

    private TreeNode build(TreeNode listNode, TreeNode root) {
        if (root == null) return listNode;
        listNode = build(listNode, root.left);
        listNode.left = root;
        listNode = root;
        return build(listNode, root.right);
    }
		
    public boolean hasNext() {
        return current != null;
    }
    public int next() {
        int res = current.val;
        current = current.left;
        return res;
    }
}