LeetCode-173-二叉搜索树迭代器

解题思路

使用一个ArrayList保存按中序遍历顺序的节点,因为二叉搜索树按中序遍历的结果是升序的
再定义一个index = -1,先判断索引index + 1 < nodes.size()是否合法
合法则next()返回ArrayList对应索引index + 1的元素

代码

class BSTIterator {
    ArrayList<Integer> nodes = new ArrayList<>();
    int index = -1;

    public BSTIterator(TreeNode root) {
        dfs(root);
    }

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

    public boolean hasNext() {
        return index + 1 < nodes.size();
    }

    private void dfs(TreeNode root) {
        if (root == null) {
            return;
        }
        dfs(root.left);
        nodes.add(root.val);
        dfs(root.right);
    }
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容