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() 被调用时，二维向量总是存在至少一个后续元素。

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/flatten-2d-vector
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

class Vector2D {
	vector<vector<int>> arr;
	int r = 0, c = 0, m;
public:
    Vector2D(vector<vector<int>>& v) {
    	arr = v;
        m = arr.size();
    }
    
    int next() {
        while(c == arr[r].size())//跳过没有剩余元素的行
    	{
    		r++;
    		c = 0;
    	}
    	return arr[r][c++];
    }
    
    bool hasNext() {
        while(r < m && c == arr[r].size())
    	{	//跳过没有剩余元素的行
    		r++;
    		c = 0;
    	}
    	return r < m-1 || (r == m-1 && c < arr[m-1].size());
    }
};

72 ms 21.9 MB

---

我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号（Michael阿明），一起加油、一起学习进步！