zigzagiterator.py

class ZigZagIterator:
    def __init__(self, v1, v2):
        """
        Initialize your data structure here.
        :type v1: List[int]
        :type v2: List[int]
        """
        self.queue = [_ for _ in (v1, v2) if _]
        print(self.queue)

    def next(self):
        """
        :rtype: int
        """
        v = self.queue.pop(0)
        ret = v.pop(0)
        if v:
            self.queue.append(v)
        return ret

    def has_next(self):
        """
        :rtype: bool
        """
        if self.queue:
            return True
        return False


l1 = [1, 2]
l2 = [3, 4, 5, 6]
it = ZigZagIterator(l1, l2)
while it.has_next():
    print(it.next())