给出n个一维的向量,请你实现一个迭代器,交替返回它们中间的元素。
示例:
输入: v1 = [1,2] v2 = [3,4,5,6] 输出: [1,3,2,4,5,6]
import java.util.Iterator;
import java.util.LinkedList;
public class UpDownIterator implements Iterator {
LinkedList<Integer> queue = new LinkedList<>();
public UpDownIterator(int[] ... arrs) {
// 计算最大的数组长度
int maxLength = 0;
for (int j = 0; j< arrs.length; j++) {
maxLength = Math.max(arrs[j].length,maxLength);
}
// 使用最大的数组长度作为边界,竖着开始遍历二维数组
for (int i = 0; i < maxLength; i++) {
for (int j = 0; j< arrs.length; j++) {
// i 在小于 一维数组的长度情况下,将该坐标的值放进队列中
if( i < arrs[j].length){
queue.offer(arrs[j][i]);
}
}
}
}
public static void main(String[] args) {
int[] v1 = { 1,2};
int[] v2 = { 3,4,0};
int[] v3 = { 5,6};
UpDownIterator upDownIterator = new UpDownIterator(v1,v2,v3);
while(upDownIterator.hasNext()){
System.out.println(upDownIterator.next());
}
}
@Override
public boolean hasNext() {
return !queue.isEmpty();
}
@Override
public Object next() {
return queue.pop();
}
}