281. Zigzag Iterator

最后更新

二刷

11-Jan-2017

第一种方式是各种pointer标记轮到哪个LIST，在那个LIST是什么index，比较繁琐。

第二种方式是用Queue < List < Integer > >,每次拿出最前面的list，拿出里面的element，然后如果此时list不是空的，从屁眼塞回原来的Queue里。

Time Complexity:

init: O(1)

hasNext: O(1)

next(): O(1)

Space: O(n)

public class ZigzagIterator {

Queue<List<Integer>> q = new LinkedList<>();
public ZigzagIterator(List<Integer> v1, List<Integer> v2) {
if (!v1.isEmpty()) {
q.add(v1);
}
if (!v2.isEmpty()) {
q.add(v2);
}
}

public int next() {
List<Integer> tempList = q.poll();
int res = tempList.remove(0);
if (!tempList.isEmpty()) {
q.add(tempList);
}
return res;
}

public boolean hasNext() {
return !q.isEmpty();
}
}

follow up: K lists

没啥变化吧。。按顺序塞屁眼，出来的时候还是这个顺序。

一刷

08-Oct-2017

我是直接按K个来做的，记录总数和已读数量判断hasNext()；

一个INDEX，每次mod K得到哪个LIST，/k得到具体INDEX位置，当然如果INDEX位置超过当前LIST的长度，就看下一个LIST。。

public class ZigzagIterator
{
int total = 0;
int cur = 0;
int have = 0;
List<Integer> l1;
List<Integer> l2;
public ZigzagIterator(List<Integer> v1, List<Integer> v2)
{
total = v1.size() + v2.size();
length = Math.max(v1.size(),v2.size())*2;
this.l1 = v1;
this.l2 = v2;
}

public int next()
{
have++;
while(true)
{
if(cur % 2 == 0)
{
if(cur/2 < l1.size())
{
cur++;
return l1.get((cur-1)/2);
}
else cur++;
}
else
{
if(cur/2 < l2.size())
{
cur++;
return l2.get((cur-1)/2);
}
else cur++;
}
}

}

public boolean hasNext()
{
return (have < total);
}
}

Discussion Board提供了用Queue的思路而不用记录INDEX的方法，其实一开始我想到都添加到一个LIST里面，但是会修改原来的LIST，毕竟要删除掉，否则还要记录每个的INDEX，就没意义了。。

下面这个QUEUE是不需要记录INDEX的方法 ..

public class ZigzagIterator
{
Queue<List<Integer>> q = new LinkedList<List<Integer>>();
public ZigzagIterator(List<Integer> v1, List<Integer> v2)
{
if(!v1.isEmpty())
q.add(v1);
if(!v2.isEmpty())
q.add(v2);
}

public int next()
{
List<Integer> tempList = q.poll();
int res = tempList.remove(0);
if(!tempList.isEmpty()) q.add(tempList);
return res;

}

public boolean hasNext()
{
return (!q.isEmpty());
}
}