LeetCode-284. 顶端迭代器-Java-medium

题目链接

法一

/**
 * 法一
 * （1）构造函数传进来的迭代器已经有了next和haseNext函数，需要增加peek函数
 * （2）可以加一个缓冲变量cache，记录当前要返回的值
 * （3）AtomicReference对应普通的对象引用，它可以保证你在修改对象引用时的线程安全性
 * （4）下面是一个线程安全的实现，把Integer改成泛型就可以支持其他类型
 */
public class Solution284 implements Iterator<Integer> {

    private AtomicReference<Integer> cache = new AtomicReference<>(null);
    private Iterator<Integer> iter;

    /**
     * 使用指定整数迭代器iterator初始化迭代器
     *
     * @param iterator
     */
    public Solution284(Iterator<Integer> iterator) {
        this.iter = iterator;
        cache.set(iter.hasNext() ? iter.next() : null);
    }

    /**
     * 返回数组中的下一个元素，但不移动指针
     *
     * @return
     */
    public Integer peek() {
        return cache.get(); // 将缓冲变量直接返回
    }

    /**
     * 返回数组中的下一个元素，并将指针移动到下个元素处
     *
     * @return
     */
    @Override
    public Integer next() { // 需要更新缓冲变量，然后将之前的缓冲变量返回
        return cache.getAndSet(iter.hasNext() ? iter.next() : null); // 以原子方式设置为给定值并返回旧值
    }

    /**
     * 如果数组中存在下一个元素，返回true；否则，返回false
     *
     * @return
     */
    @Override
    public boolean hasNext() {
        return cache.get() != null;
    }
    
}

本地测试

        /**
         * 284. 顶端迭代器
         */
        lay.showTitle(284);
        List<Integer> list284 = Arrays.asList(1, 2, 3); // 创建集合
        Iterator<Integer> iterator284 = list284.iterator(); // 获取迭代器
        Solution284 sol284 = new Solution284(iterator284);
        System.out.println(list284);
        System.out.println(sol284.next());
        System.out.println(sol284.peek());
        System.out.println(sol284.next());
        System.out.println(sol284.next());
        System.out.println(sol284.hasNext());