题目描述

请设计并实现一个能够展开二维向量的迭代器。该迭代器需要支持 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() 被调用时，二维向量总是存在至少一个后续元素。

算法1

使用cur和index表示当前遍历到的那个vector以及当前的下标
注意可能会存在空的vector；

C++ 代码

class Vector2D {
    vector<vector<int>> ans;
    int cur;
    int index;
public:
    Vector2D(vector<vector<int>>& v) {
        ans = v;
        cur = 0;
        index = 0;
    }

    int next() {
        while (index==ans[cur].size()) {
            cur++;
            index = 0;
        }
        int res= ans[cur][index];
        index++;
        return res;
    }

    bool hasNext() {
        while (cur < ans.size() && index == ans[cur].size()) {
            cur++;
            index = 0;
        }
        if (cur == ans.size() - 1 && index == ans[cur].size()) return false;
        return true;
    }
};