2021-10-05 [Mnsx_x]刷题笔记——顶端迭代器

[Mnsx_x]刷题笔记——顶端迭代器

请你设计一个迭代器，除了支持 hasNext 和 next 操作外，还支持 peek 操作。
实现 PeekingIterator 类：
PeekingIterator(int[] nums) 使用指定整数数组 nums 初始化迭代器。
int next() 返回数组中的下一个元素，并将指针移动到下个元素处。
bool hasNext() 如果数组中存在下一个元素，返回 true；否则，返回 false 。
int peek() 返回数组中的下一个元素，但不移动指针。

题目解析

很简单的模拟题，与java中的iterator不同点在于，拥有一个peek()方法，能够返回下一个数并且不移动迭代器

思路一——模拟

模拟Iterator的功能来完成解题，添加一个Integer来实现peek方法，调用peek时返回Integer的值，next则返回Integer的值，并且更新Integer的值为现在指针的下一位，hasNext则判断Integer的值，为null则返回false，反之则返回true

// Java Iterator interface reference:
// https://docs.oracle.com/javase/8/docs/api/java/util/Iterator.html

class PeekingIterator implements Iterator<Integer> {
	//使用Java的迭代器来处理数据
	Iterator<Integer> it;
	//使用一个Integer来实现peek的功能
	Integer next;
	public PeekingIterator(Iterator<Integer> iterator) {
	    // initialize any member here.
	    it = iterator;
		if(it.hasNext()){
			next = it.next();
		}
	}
	
    // Returns the next element in the iteration without advancing the iterator.
	public Integer peek() {
        return next;
	}
	
	// hasNext() and next() should behave the same as in the Iterator interface.
	// Override them if needed.
	@Override
	public Integer next() {
		Integer ans = next;
		next = it.hasNext() ? it.next() : null;
		return ans;	
	}
	
	@Override
	public boolean hasNext() {
	   return (next != null); 
	}
}

时间复杂度：O(1)
空间复杂度：O(1)

思路二——模拟 + 泛型

参考三叶大佬完成，使得迭代器可以支持所有类型

得益于 Java 的「泛型」设计，我们可以很轻松地支持任意类型：只需要将 Integer 修改成代指泛型的标识即可，例如 E。

class PeekingIterator implements Iterator<E> {
    Iterator<E> iter;
    E next;
	public PeekingIterator(Iterator<E> iterator) {
        iter = iterator;
        if (iter.hasNext()) next = iter.next();
	}
	
	public E peek() {
        return next;
	}
	
	@Override
	public E next() {
        E ans = next;
        next = iter.hasNext() ? iter.next() : null;
	    return ans;
	}
	
	@Override
	public boolean hasNext() {
        return next != null;
	}
}

参考文献