首页 > itarticle > [lintcode] problem 601 – flatten 2d vector

[lintcode] problem 601 – flatten 2d vector

admin 11月 13, 2020 0

Implement an iterator to flatten a 2d vector.

Example

No.1

Input:[[1,2],[3],[4,5,6]]

Output:[1,2,3,4,5,6]

No.2

Input:[[7,9],[5]]

Output:[7,9,5]

Code

private Iterator<List<Integer>> row;
private Iterator<Integer> col;

public (List<List<Integer>> vec2d) {
    row = vec2d.iterator();
}


public Integer next() {
    if (!hasNext())
        return null;

    return col.next();
}


public boolean hasNext() {
    while ((col == null || !col.hasNext()) && row.hasNext())
        col = row.next().iterator();

    return col != null && col.hasNext();
}