每日一题 【每日一题】左旋右旋迭代器-20211204

Jack · 2021年12月04日 · 最后由 Jack 回复于 2021年12月05日 · 91 次阅读
本帖已被设为精华帖!

给出两个一维向量,实现一个迭代器,交替返回两个向量的元素

Jack 将本帖设为了精华贴 12月04日 22:25

参考代码:

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

需要 登录 后方可回复, 如果你还没有账号请点击这里 注册