给出两个一维向量,实现一个迭代器,交替返回两个向量的元素
参考代码:
class Solution: def __init__(self, v1, v2): self.queue = [v for v in (v1, v2) if v] def next(self): v = set.queue.pop(0) value = v.pop(0) if v: self.queue.append(v) return value def hasNext(self): return len(self.queue) > 0