FilesExpand file tree

 master

/

ZigzagIterator.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

41 lines (34 loc) · 928 Bytes

 master

/

ZigzagIterator.java

Copy path

Top

File metadata and controls

• Code
• Blame

41 lines (34 loc) · 928 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

32

33

34

35

36

37

38

39

40

41

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

/**

* 可以轻易拓展到k的情况

*/

public class ZigzagIterator {

private List<Iterator<Integer>> mIterators = new ArrayList<>();

private int mIndex;

public ZigzagIterator(List<Integer> v1, List<Integer> v2) {

init(v1, v2);

}

private void init(List<Integer>... lists) {

for (List<Integer> list : lists) {

if (!list.isEmpty()) {

mIterators.add(list.iterator());

}

}

}

public int next() {

mIndex %= mIterators.size();

Iterator<Integer> iterator = mIterators.get(mIndex);

int n = iterator.next();

if (!iterator.hasNext()) {

mIterators.remove(iterator);

} else {

mIndex++;

}

return n;

}

public boolean hasNext() {

return !mIterators.isEmpty();

}

}