LeetCode刷题实战251：展开二维向量
算法的重要性，我就不多说了吧，想去大厂，就必须要经过基础知识和业务逻辑面试+算法面试。所以，为了提高大家的算法能力，这个公众号后续每天带大家做一道算法题，题目就从LeetCode上面选 ！
今天和大家聊的问题叫做 展开二维向量，我们先来看题面：
https://leetcode-cn.com/problems/flatten-2d-vector/
Implement an iterator to flatten a 2d vector.
请设计并实现一个能够展开二维向量的迭代器。该迭代器需要支持 next 和 hasNext 两种操作。
示例
Input: 2d vector =
[
[1,2],
[3],
[4,5,6]
]
Output: [1,2,3,4,5,6]
解题
需要维护两个动态迭代器row_it和col_it，分别表示当前位置的行和列。但是由于我们不存储vec2d本身，所以我们还需要另外一个静态迭代器row_end，表示整个二维数组的末尾。这里需要特别注意的是在判断hasNext的时候，要处理那种跳过空行的情况，请见下面的代码实现。
class Vector2D {
public:
Vector2D(vector>& vec2d) {
if (vec2d.size() == 0) {
return;
}
row_it = vec2d.begin(), row_end = vec2d.end();
col_it = (*row_it).begin();
}
int next() {
return *col_it++;
}
bool hasNext() {
while (row_it != row_end && col_it == (*row_it).end()) {
++row_it;
col_it = (*row_it).begin();
}
return row_it != row_end;
}
private:
vector>::iterator row_it, row_end;
vector::iterator col_it;
};
好了，今天的文章就到这里，如果觉得有所收获，请顺手点个在看或者转发吧，你们的支持是我最大的动力 。
上期推文：
LeetCode1-240题汇总，希望对你有点帮助！
LeetCode刷题实战241：为运算表达式设计优先级
LeetCode刷题实战242：有效的字母异位词
LeetCode刷题实战243：最短单词距离
LeetCode刷题实战244：最短单词距离 II
LeetCode刷题实战245：最短单词距离 III
LeetCode刷题实战246：中心对称数
LeetCode刷题实战247：中心对称数II
LeetCode刷题实战248：中心对称数III
LeetCode刷题实战249：移位字符串分组
LeetCode刷题实战250：统计同值子树
