跳转至

LeetCode: 251. 展开二维向量

1、题目描述

请设计并实现一个能够展开二维向量的迭代器。该迭代器需要支持 next 和 hasNext 两种操作。、

示例:

Vector2D iterator = new Vector2D([[1,2],[3],[4]]);

iterator.next(); // 返回 1
iterator.next(); // 返回 2
iterator.next(); // 返回 3
iterator.hasNext(); // 返回 true
iterator.hasNext(); // 返回 true
iterator.next(); // 返回 4
iterator.hasNext(); // 返回 false

注意:

  • 请记得 重置 在 Vector2D 中声明的类变量(静态变量),因为类变量会 在多个测试用例中保持不变,影响判题准确。请 查阅 这里。

  • 你可以假定 next() 的调用总是合法的,即当 next() 被调用时,二维向量总是存在至少一个后续元素。

2、解题思路

  • 将数组展开成一维
class Vector2D:

    def __init__(self, v: List[List[int]]):

        func = lambda x: [y for l in x for y in func(l)] if type(x) is list else [x]
        self.vector = func(v)
        self.vector_pos = 0

    def next(self) -> int:
        temp = self.vector[self.vector_pos]
        self.vector_pos += 1
        return temp

    def hasNext(self) -> bool:

        return self.vector_pos < len(self.vector)


# Your Vector2D object will be instantiated and called as such:
# obj = Vector2D(v)
# param_1 = obj.next()
# param_2 = obj.hasNext()