​LeetCode刷题实战251:展开二维向量

程序IT圈

共 2261字,需浏览 5分钟

 ·

2021-05-05 07:02

算法的重要性,我就不多说了吧,想去大厂,就必须要经过基础知识和业务逻辑面试+算法面试。所以,为了提高大家的算法能力,这个公众号后续每天带大家做一道算法题,题目就从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<vector<int>>& 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<vector<int>>::iterator row_it, row_end;
    vector<int>::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:统计同值子树


浏览 5
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报