FilesExpand file tree

 master

/

zigzagiterator.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

31 lines (28 loc) · 645 Bytes

 master

/

zigzagiterator.py

Copy path

Top

File metadata and controls

• Code
• Blame

31 lines (28 loc) · 645 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

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())