281. Zigzag Iterator

Zigzag迭代器实现

最新推荐文章于 2021-02-10 05:47:32 发布

转载最新推荐文章于 2021-02-10 05:47:32 发布 · 63 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zmyvszk/p/5622494.html

本文介绍了一种名为ZigzagIterator的类实现，该类能够从两个整数列表中以锯齿形方式取出元素。通过使用LinkedList保存每个列表的迭代器，并在每次调用next方法时交替获取不同列表中的元素。

/*
 * 281. Zigzag Iterator
 * 2016-6-27 by Mingyang
 * 这里就利用了list里面的Iterator的性质，不断地交替取出Iterator，并且取下一个值
 */
 class ZigzagIterator {
     LinkedList<Iterator> list;
        public ZigzagIterator(List<Integer> v1, List<Integer> v2) {
            list = new LinkedList<Iterator>();
            if(!v1.isEmpty()) list.add(v1.iterator());
            if(!v2.isEmpty()) list.add(v2.iterator());
        }
        public int next() {
            Iterator poll = list.remove();
            int result = (Integer)poll.next();
            if(poll.hasNext()) list.add(poll);
            return result;
        }
        public boolean hasNext() {
            return !list.isEmpty();
        }
    }